fixed UI

app.py

@@ -111,9 +111,9 @@ def handle_geojson_upload(file):
 def retrieve_paragraphs(query):
     """Connect to retriever and retrieve paragraphs"""
 
-    try:        
+    try:   
         # Call the API with the uploaded file
-        client = Client("https://giz-eudr-retriever.hf.space/")
+        client = Client("https://giz-chatfed-retriever.hf.space/")
         result = client.predict(
                 query=query,
                 reports_filter="",
@@ -122,7 +122,7 @@ def retrieve_paragraphs(query):
                 year_filter="",
                 api_name="/retrieve"
         )    
-
+        print(client)
         return (
             results,  
             gr.update(visible=True),  # Keep status visible
@@ -152,11 +152,65 @@ def finish_chat():
     """Finish chat and reset input"""
     return gr.update(interactive=True, value="")
 
-async def chat_response(query, history, method, country, uploaded_file):
+# async def chat_response(query, history, method, country, uploaded_file):
+
+#     """Generate chat response based on method and inputs"""
+    
+#     # Validate inputs based on method
+#     if method == "Upload GeoJSON":
+#         if uploaded_file is None:
+#             warning_message = "⚠️ **No GeoJSON file uploaded.** Please upload a GeoJSON file first."
+#             history[-1] = (query, warning_message)
+#             yield history, ""
+#             return
+#     else:  # "Talk to Reports"
+#         if not country:
+#             warning_message = "⚠️ **No country selected.** Please select a country to analyze reports."
+#             history[-1] = (query, warning_message)
+#             yield history, ""
+#             return
+    
+#     # Get the formatted statistics if a file was just uploaded
+#     if method == "Upload GeoJSON" and uploaded_file:
+#         try:
+#             stats_result = handle_geojson_upload(uploaded_file)
+#             formatted_stats = stats_result[0]  # Get the formatted statistics
+#             response = formatted_stats
+#         except Exception as e:
+#             response = f"Error processing file: {str(e)}"
+#     elif method == "Upload GeoJSON":
+#             response = f"Based on your uploaded GeoJSON file, I can help you analyze the deforestation patterns and EUDR compliance aspects in your area of interest. Your question: '{query}' is being processed against the geographic data you provided."
+    
+#     # Talk to report
+#     else: 
+#         try: 
+#             # response = f"Based on EUDR reports for {country}, I can help you understand deforestation patterns and compliance requirements. Your question: '{query}' is being analyzed against our {country} database."
+            
+#             # Retrieve info
+#             print("retrieve info")
+#             retrieved_info = retrieve_paragraphs(query)[0]
+#             response = retrieved_info
+
+#         except Exception as e:
+#             response = f"Error retrieving information: {str(e)}"
+
 
+   
+#     #     else:
+#     #         response = f"Based on EUDR reports for {country}, I can help you understand deforestation patterns and compliance requirements. Your question: '{query}' is being analyzed against our {country} database."
+    
+#     # Simulate streaming response
+#     words = response.split()
+#     for word in words:
+#         history[-1] = (query, " ".join(words[:words.index(word)+1]))
+#         yield history, "**Sources:** Sample source documents would appear here..."
+#         await asyncio.sleep(0.05)
+
+
+async def chat_response(query, history, method, country, uploaded_file):
     """Generate chat response based on method and inputs"""
     
-    # Validate inputs based on method
+    # Validate inputs
     if method == "Upload GeoJSON":
         if uploaded_file is None:
             warning_message = "⚠️ **No GeoJSON file uploaded.** Please upload a GeoJSON file first."
@@ -170,39 +224,27 @@ async def chat_response(query, history, method, country, uploaded_file):
             yield history, ""
             return
     
-    # Get the formatted statistics if a file was just uploaded
+    # Handle GeoJSON upload → put formatted stats into chat
     if method == "Upload GeoJSON" and uploaded_file:
         try:
-            stats_result = handle_geojson_upload(uploaded_file)
-            formatted_stats = stats_result[0]  # Get the formatted statistics
+            formatted_stats = handle_geojson_upload(uploaded_file)  # returns plain text
             response = formatted_stats
         except Exception as e:
             response = f"Error processing file: {str(e)}"
     
-    # Talk to report
+    # Handle "Talk to Reports"
     else: 
         try: 
-            response = f"Based on EUDR reports for {country}, I can help you understand deforestation patterns and compliance requirements. Your question: '{query}' is being analyzed against our {country} database."
-            
-            # Retrieve info
+            response = f"Based on EUDR reports for {country}, I can help you understand deforestation patterns and compliance requirements. Your question: '{query}' is being analyzed..."
             retrieved_info = retrieve_paragraphs(query)[0]
             response = retrieved_info
-
         except Exception as e:
             response = f"Error retrieving information: {str(e)}"
 
-
-    # else:
-    #     # Default response for other queries
-    #     if method == "Upload GeoJSON":
-    #         response = f"Based on your uploaded GeoJSON file, I can help you analyze the deforestation patterns and EUDR compliance aspects in your area of interest. Your question: '{query}' is being processed against the geographic data you provided."
-    #     else:
-    #         response = f"Based on EUDR reports for {country}, I can help you understand deforestation patterns and compliance requirements. Your question: '{query}' is being analyzed against our {country} database."
-    
-    # Simulate streaming response
+    # Stream response word by word into the chat
     words = response.split()
-    for word in words:
-        history[-1] = (query, " ".join(words[:words.index(word)+1]))
+    for i in range(len(words)):
+        history[-1] = (query, " ".join(words[:i+1]))
         yield history, "**Sources:** Sample source documents would appear here..."
         await asyncio.sleep(0.05)
 
@@ -475,10 +517,16 @@ with gr.Blocks(title="EUDR Q&A", theme=theme, css=custom_css) as demo:
     )
     
     # File upload - automatically process when file is uploaded
+    # uploaded_file.change(
+    #     fn=handle_geojson_upload,
+    #     inputs=[uploaded_file],
+    #     outputs=[upload_status, upload_status, results_table]
+    # )
+
     uploaded_file.change(
-        fn=handle_geojson_upload,
-        inputs=[uploaded_file],
-        outputs=[upload_status, upload_status, results_table]
+    lambda _: None,
+    inputs=[uploaded_file],
+    outputs=[]
     )
     
     # Chat functionality