added UI changes

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import time
 import pandas as pd
+import asyncio
 from uuid import uuid4
 from gradio_client import Client, handle_file
 
@@ -23,6 +24,40 @@ SAMPLE_QUESTIONS = {
     ]
 }
 
+def format_whisp_statistics(df):
+    """Format WhispAPI statistics into a standardized, readable text"""
+    try:
+        # Extract required indicators from API response
+        country = df['Country'].iloc[0]
+        admin_level = df['Admin_Level_1'].iloc[0]
+        area = round(df['Area'].iloc[0], 2)
+        risk_level = df['risk_level'].iloc[0] if 'risk_level' in df.columns else "Not available"
+        risk_pcrop = df['risk_pcrop'].iloc[0] if 'risk_pcrop' in df.columns else "Not available"
+        risk_acrop = df['risk_acrop'].iloc[0] if 'risk_acrop' in df.columns else "Not available"
+        risk_timber = df['risk_timber'].iloc[0] if 'risk_timber' in df.columns else "Not available"
+        deforestation_after_2020 = df['TMF_def_after_2020'].iloc[0] if 'TMF_def_after_2020' in df.columns else 0
+        
+        # Format the text output
+        output = f"""📊 Analysis Results for Your Plot
+
+Plot Information:
+- Country: {country}
+- Administrative Region: {admin_level}
+- Total Area: {area} hectares
+
+Risk Assessment:
+- Overall Risk Level: {risk_level}
+- Permanent Crop Risk: {risk_pcrop}
+- Annual Crop Risk: {risk_acrop}
+- Timber Risk: {risk_timber}
+
+Deforestation Analysis:
+- Deforestation after 2020: {round(deforestation_after_2020, 2)} hectares
+"""
+        return output
+    except Exception as e:
+        return f"Error formatting statistics: {str(e)}"
+
 def handle_geojson_upload(file):
     """Handle GeoJSON file upload and call WHISP API"""
     if file is not None:
@@ -39,10 +74,13 @@ def handle_geojson_upload(file):
             # Convert result to DataFrame
             df = pd.DataFrame(result['data'], columns=result['headers'])
             
+            # Format statistics into readable text
+            formatted_stats = format_whisp_statistics(df)
+            
             return (
-                "✅ GeoJSON file processed successfully! Analysis results are displayed below.",
-                gr.update(visible=True),  # upload_status
-                gr.update(value=df, visible=True)  # results_table
+                formatted_stats,  # Keep formatted statistics for chat
+                gr.update(visible=True),  # Keep status visible
+                gr.update(visible=False)  # Always hide results table
             )
             
         except Exception as e:
@@ -59,15 +97,12 @@ def handle_geojson_upload(file):
             gr.update(visible=False)   # results_table
         )
 
-
-def retrieve_paragraphs(query):
-
+def retrieve_paragraphs(file):
     """Connect to retriever and retrieve paragraphs"""
-    
     if file is not None:
         try:
             # Initialize WHISP API client
-            client = Client("https://giz-eudr-retriever.hf.space/")
+            client = Client("https://g...content-available-to-author-only...f.space/")
             
             # Call the API with the uploaded file
             result = client.predict(
@@ -75,27 +110,14 @@ def retrieve_paragraphs(query):
                 api_name="/retrieve"
             )
             
-            return (
-                "These are the most relevant findings.",
-                gr.update(visible=True),  # upload_status
-                gr.update(value=results, visible=True)  # results_table
-            )
+            # Convert result to DataFrame
+            df = pd.DataFrame(result['data'], columns=result['headers'])
+            return df
             
         except Exception as e:
-            error_msg = f"❌ Error creating a response {str(e)}"
-            return (
-                error_msg,
-                gr.update(visible=True),  # upload_status
-                gr.update(visible=False)  # results_table
-            )
-    
-    else:
-        return (
-            "",
-            gr.update(visible=False),  # upload_status
-            gr.update(visible=False)   # results_table
-         )
-
+            print(f"Error retrieving paragraphs: {str(e)}")
+            return None
+    return None
 
 def start_chat(query, history):
     """Start a new chat interaction"""
@@ -123,18 +145,25 @@ async def chat_response(query, history, method, country, uploaded_file):
             yield history, ""
             return
     
-    # Simulate processing time
-    response = ""
-    if method == "Upload GeoJSON":
-        full_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."
+    # 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)}"
     else:
-        full_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."
+        # 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
-    words = full_response.split()
+    words = response.split()
     for word in words:
-        response += word + " "
-        history[-1] = (query, response)
+        history[-1] = (query, " ".join(words[:words.index(word)+1]))
         yield history, "**Sources:** Sample source documents would appear here..."
         await asyncio.sleep(0.05)
 
@@ -469,4 +498,4 @@ with gr.Blocks(title="EUDR Q&A", theme=theme, css=custom_css) as demo:
 
 # Launch the app
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()