Update app.py

app.py

@@ -65,84 +65,60 @@ def stream_chat_with_rag(
     return answer
 
 
-def format_answer_string(answer: str):
-    """
-    This function extracts and formats the assistant's response before document metadata.
-    Anything after the marker `[(` (where documents are listed) is ignored.
-    """
-    # Step 1: Split the response at the start of the document metadata
-    split_marker = "[("
-    if split_marker in answer:
-        # Everything before the marker is the relevant answer
-        answer_before_docs = answer.split(split_marker)[0]
-    else:
-        # If no documents metadata, return the entire answer
-        answer_before_docs = answer
+# def format_answer_string(answer: str):
+#     """
+#     This function extracts and formats the assistant's response before document metadata.
+#     Anything after the marker `[(` (where documents are listed) is ignored.
+#     """
+#     # Step 1: Split the response at the start of the document metadata
+#     split_marker = "[("
+#     if split_marker in answer:
+#         # Everything before the marker is the relevant answer
+#         answer_before_docs = answer.split(split_marker)[0]
+#     else:
+#         # If no documents metadata, return the entire answer
+#         answer_before_docs = answer
     
-    # Step 2: Clean up formatting by replacing escaped newline characters
-    formatted_answer = answer_before_docs.replace("\\n", "\n").strip()
+#     # Step 2: Clean up formatting by replacing escaped newline characters
+#     formatted_answer = answer_before_docs.replace("\\n", "\n").strip()
     
-    # Step 3: Remove potential starting and ending artifacts like (' and ,) if present
-    if formatted_answer.startswith("(\"") and formatted_answer.endswith("\","):
-        formatted_answer = formatted_answer[2:-2].strip()
+#     # Step 3: Remove potential starting and ending artifacts like (' and ,) if present
+#     if formatted_answer.startswith("(\"") and formatted_answer.endswith("\","):
+#         formatted_answer = formatted_answer[2:-2].strip()
 
-    # Optional: Add a prefix for clarity
-    formatted_answer = "Co-Pilot: " + formatted_answer
+#     # Optional: Add a prefix for clarity
+#     formatted_answer = "Co-Pilot: " + formatted_answer
     
-    return formatted_answer
-
-def format_relevant_documents(relevant_docs: list):
-    """
-    This function formats the relevant document metadata and content for readable output.
-    It extracts the heading, page number, and a snippet of the content from each document.
-    """
-    formatted_docs = "Relevant Documents:\n\n"
+#     return formatted_answer
+
+# def format_relevant_documents(relevant_docs: list):
+#     """
+#     This function formats the relevant document metadata and content for readable output.
+#     It extracts the heading, page number, and a snippet of the content from each document.
+#     """
+#     formatted_docs = "Relevant Documents:\n\n"
     
-    for idx, (doc, score) in enumerate(relevant_docs):
-        # Extract the relevant metadata
-        heading = doc.metadata.get('heading', 'Unnamed Document')
-        page_number = int(doc.metadata.get('page_number', -1))
-        source = doc.metadata.get('source', 'Unknown Source')
-        confidence = round(score, 4)  # Rounding the score for cleaner output
+#     for idx, (doc, score) in enumerate(relevant_docs):
+#         # Extract the relevant metadata
+#         heading = doc.metadata.get('heading', 'Unnamed Document')
+#         page_number = int(doc.metadata.get('page_number', -1))
+#         source = doc.metadata.get('source', 'Unknown Source')
+#         confidence = round(score, 4)  # Rounding the score for cleaner output
         
-        # Add the formatted details to the output string
-        formatted_docs += f"Document {idx + 1}:\n"
-        formatted_docs += f"  - Heading: {heading}\n"
-        formatted_docs += f"  - Page Number: {page_number}\n"
-        formatted_docs += f"  - Source: {source}\n"
-        formatted_docs += f"  - Confidence Score: {confidence}\n"
+#         # Add the formatted details to the output string
+#         formatted_docs += f"Document {idx + 1}:\n"
+#         formatted_docs += f"  - Heading: {heading}\n"
+#         formatted_docs += f"  - Page Number: {page_number}\n"
+#         formatted_docs += f"  - Source: {source}\n"
+#         formatted_docs += f"  - Confidence Score: {confidence}\n"
         
-        # Optionally include a snippet from the content
-        content_snippet = doc.page_content[:200]  # Take the first 200 characters for preview
-        formatted_docs += f"  - Content Snippet: {content_snippet}...\n\n"
+#         # Optionally include a snippet from the content
+#         content_snippet = doc.page_content[:200]  # Take the first 200 characters for preview
+#         formatted_docs += f"  - Content Snippet: {content_snippet}...\n\n"
 
-    return formatted_docs.strip()
+#     return formatted_docs.strip()
 
 
-# # ------------------------------------- Core CNI APP ----------------------------------------------------------------------------------
-# # Function to handle PDF processing API call
-# def process_pdf(pdf_file, client_name):
-#     return client.predict(
-#         pdf_file=handle_file(pdf_file),
-#         #client_name=client_name,  # Hardcoded client name
-#         api_name="/process_pdf"
-#     )[1]  # Return only the result string
-
-# # # Function to handle search API call
-# # def search_api(query):
-# #     return client.predict(query=query, api_name="/search_with_confidence")
-
-# # Function to handle RAG API call
-# def rag_api(question, client_name):
-#     return client.predict(question=question,selected_document=client_name, api_name="/get_answer")
-
-# def delete_index():
-#     result = client.predict(api_name="/delete_index")
-#     return result
-
-
-
-#-------------------------------------- UX & Gradio -------------------------------------------------------------------------------
 
 # CSS for custom styling
 CSS = """
@@ -156,7 +132,7 @@ CSS = """
 
 
 # Title for the application
-TITLE = "<h1 style='text-align:center;'>CNI RAG AGENTIC v0.2</h1>"
+TITLE = "<h1 style='text-align:center;'>Reddit Election Q&A agent v0.1</h1>"
 
 # Create the Gradio Blocks interface
 with gr.Blocks(css=CSS) as demo: