Update app.py

app.py

@@ -44,7 +44,12 @@ def stream_chat_with_rag(
     # Call the API with the user's process_query
     question = message
     #answer = client.predict(question=question, api_name="/run_graph")
-    answer = client.predict(question=question,selected_document=client_name, api_name="/process_query")
+    answer = client.predict(
+		query="Hello!!",
+		context="Hello!!",
+		election_year="2016 Election",
+		api_name="/process_query"
+    )
 
     # Debugging: Print the raw response
     print("Raw answer from API:")
@@ -114,33 +119,28 @@ def format_relevant_documents(relevant_docs: list):
     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
+# # ------------------------------------- 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 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")
+# # 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
+# def delete_index():
+#     result = client.predict(api_name="/delete_index")
+#     return result
 
 
-#-------------------------------------- View PDF ----------------------------------------------------------------------------------
-def view_pdf(pdf):
-    result = client.predict(api_name="/view_pdf_name")
-    return result
-
 
 #-------------------------------------- UX & Gradio -------------------------------------------------------------------------------
 
@@ -168,7 +168,7 @@ with gr.Blocks(css=CSS) as demo:
             chatbot=chatbot,
             additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
             additional_inputs=[
-                gr.Dropdown(client_name,value="primo",label="Select Client", render=False,allow_custom_value=True),
+                gr.Dropdown(client_name,value="2016 Election",label="Select Election year", render=False,allow_custom_value=True),
                 gr.Textbox(
                     value="You are an expert assistant",
                     label="System Prompt",
@@ -234,34 +234,34 @@ with gr.Blocks(css=CSS) as demo:
         )
 
 
-    with gr.Tab("Process PDF"):
-        pdf_input = gr.File(label="Upload PDF File")
-        #select_client_dropdown = gr.Dropdown(client_name, value="rosariarossi", label="Select or Type Client", allow_custom_value=True)
-        pdf_output = gr.Textbox(label="PDF Result", interactive=False)
+    # with gr.Tab("Process PDF"):
+    #     pdf_input = gr.File(label="Upload PDF File")
+    #     #select_client_dropdown = gr.Dropdown(client_name, value="rosariarossi", label="Select or Type Client", allow_custom_value=True)
+    #     pdf_output = gr.Textbox(label="PDF Result", interactive=False)
     
-        pdf_button = gr.Button("Process PDF")
-        pdf_button.click(
-            process_pdf,
-            inputs=[pdf_input],  # Pass both PDF and client name is not required 
-            outputs=pdf_output
-        )
-
-    with gr.Tab("Answer with RAG"):
-        question_input = gr.Textbox(label="Enter Question for RAG")
-        answer_with_rag_select_client_dropdown = gr.Dropdown(client_name, value="primo", label="Select or Type Client", allow_custom_value=True)
-        rag_output = gr.Textbox(label="RAG Answer Result", interactive=False)
-
-        rag_button = gr.Button("Get Answer")
-        rag_button.click(
-            rag_api,
-            inputs=[question_input,answer_with_rag_select_client_dropdown ],
-            outputs=rag_output
-        )
-    with gr.Tab(label="Manage Files"):
-        with gr.Column():
-            delete_index_button = gr.Button("Delete All Files")
-            delete_index_textout = gr.Textbox(label="Deleted Files and Refresh Result")
-            delete_index_button.click(fn=delete_index, inputs=[],outputs=[delete_index_textout])
+    #     pdf_button = gr.Button("Process PDF")
+    #     pdf_button.click(
+    #         process_pdf,
+    #         inputs=[pdf_input],  # Pass both PDF and client name is not required 
+    #         outputs=pdf_output
+    #     )
+
+    # with gr.Tab("Answer with RAG"):
+    #     question_input = gr.Textbox(label="Enter Question for RAG")
+    #     answer_with_rag_select_client_dropdown = gr.Dropdown(client_name, value="primo", label="Select or Type Client", allow_custom_value=True)
+    #     rag_output = gr.Textbox(label="RAG Answer Result", interactive=False)
+
+    #     rag_button = gr.Button("Get Answer")
+    #     rag_button.click(
+    #         rag_api,
+    #         inputs=[question_input,answer_with_rag_select_client_dropdown ],
+    #         outputs=rag_output
+    #     )
+    # with gr.Tab(label="Manage Files"):
+    #     with gr.Column():
+    #         delete_index_button = gr.Button("Delete All Files")
+    #         delete_index_textout = gr.Textbox(label="Deleted Files and Refresh Result")
+    #         delete_index_button.click(fn=delete_index, inputs=[],outputs=[delete_index_textout])
 
 # Launch the app
 if __name__ == "__main__":