Update app.py

app.py

@@ -6,12 +6,8 @@ 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/CNIHUB101324v10", hf_token=HF_TOKEN)
+client = Client("on1onmangoes/cnianswer101724v3", hf_token=HF_TOKEN)
 
-#client = Client("on1onmangoes/CNIHUB101524v11", hf_token=HF_TOKEN)
-
-# Update the conversation history within the function.
-# Return the updated history along with any other required outputs.
 
 client_name = ['rosariarossi','bianchifiordaliso','lorenzoverdi','lucia', 'quarto4', 'quinto5', 'secondo6', 'sesto6', 'settimo7','ottavo8','nono9']
 
@@ -45,7 +41,7 @@ def stream_chat_with_rag(
 
     # Call the API with the user's message
     question = message
-    answer = client.predict(question=question, api_name="/answer_with_rag")
+    answer = client.predict(question=question, api_name="/run_graph")
 
     # Debugging: Print the raw response
     print("Raw answer from API:")
@@ -119,17 +115,17 @@ def format_relevant_documents(relevant_docs: list):
 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_pdf2"
+        #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 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):
-    return client.predict(question=question, api_name="/answer_with_rag")
+    return client.predict(question=question, api_name="/run_graph")
 
 # CSS for custom styling
 CSS = """
@@ -139,7 +135,7 @@ CSS = """
 """
 
 # Title for the application
-TITLE = "<h1 style='text-align:center;'>CNI RAG QA v0</h1>"
+TITLE = "<h1 style='text-align:center;'>CNI RAG AGENTIC v0.1</h1>"
 
 # Create the Gradio Blocks interface
 with gr.Blocks(css=CSS) as demo: