Update app.py

app.py

@@ -6,12 +6,7 @@ import os
 HF_TOKEN = os.getenv("HF_TOKEN")  # Replace with your actual token if not using an environment variable
 
 # Initialize the Gradio Client for the specified API
-#client = Client("on1onmangoes/CNIHUB10724v10", hf_token=HF_TOKEN)
 client = Client("on1onmangoes/CNIHUB101324v10", hf_token=HF_TOKEN)
-# on1onmangoes/CNIHUB101324v10
-
-
-# Here's how you can fix it:
 
 # Update the conversation history within the function.
 # Return the updated history along with any other required outputs.
@@ -77,53 +72,16 @@ def format_answer_string(answer: str):
     # Step 2: Clean up formatting by replacing escaped newline characters
     formatted_answer = answer_before_docs.replace("\\n", "\n").strip()
     
-    # Step 3: Ensure there are no unnecessary starting or ending characters like `(' or ,)`
-    # Instead of stripping them directly, check for artifacts of formatting
-    if formatted_answer.startswith("('") and formatted_answer.endswith("',"):
+    # 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 if needed
+    # Optional: Add a prefix for clarity
     formatted_answer = "Co-Pilot: " + formatted_answer
     
     return formatted_answer
 
 
-
-
-# this version works just issue with formatting 
-# def stream_chat_with_rag(
-#     message: str,
-#     history: list,
-#     client_name: str,
-#     system_prompt: str,
-#     num_retrieved_docs: int = 10,
-#     num_docs_final: int = 9,
-#     temperature: float = 0, 
-#     max_new_tokens: int = 1024, 
-#     top_p: float = 1.0, 
-#     top_k: int = 20, 
-#     penalty: float = 1.2,
-# ):
-#     print(f"Message: {message}")
-#     print(f"History: {history}")
-
-    
-#     # Build the conversation prompt including system prompt and history
-#     conversation = system_prompt + "\n\n" + "For Client:" + client_name
-#     for user_input, assistant_response in history:
-#         conversation += f"User: {user_input}\nAssistant: {assistant_response}\n"
-#     conversation += f"User: {message}\nAssistant:"
-#     question = message
-#     answer = client.predict(question=question, api_name="/answer_with_rag")
-#     # debug 092624
-#     print("The Answer in stream_chat_with_rag:")
-#     print(answer)
-
-#     # Update the conversation history
-#     history.append((message, answer))
-#     return answer
-
-
 # Function to handle PDF processing API call
 def process_pdf(pdf_file):
     return client.predict(
@@ -148,7 +106,7 @@ CSS = """
 """
 
 # Title for the application
-TITLE = "<h1 style='text-align:center;'>My Gradio Chat App</h1>"
+TITLE = "<h1 style='text-align:center;'>CNI RAG QA v0</h1>"
 
 # Create the Gradio Blocks interface
 with gr.Blocks(css=CSS) as demo: