Update app.py

app.py

@@ -6,6 +6,7 @@ from PIL import Image
 import io
 import tempfile
 import base64
+import time
 
 def analyze_environmental_impact(api_key, analysis_type, image=None, text_input=None, location=None, product_info=None):
     """
@@ -146,27 +147,48 @@ def analyze_environmental_impact(api_key, analysis_type, image=None, text_input=
         ]
     
     # Make API call
-    try:
-        completion = client.chat.completions.create(
-            extra_headers={
-                "HTTP-Referer": "https://environmental-impact-analyzer.app",
-                "X-Title": "Smart Environmental Impact Analyzer",
-            },
-            model="google/gemini-2.5-pro-exp-03-25:free",
-            messages=messages
-        )
-        
-        # Check if completion and choices exist before accessing
-        if completion and hasattr(completion, 'choices') and completion.choices and len(completion.choices) > 0:
-            if hasattr(completion.choices[0], 'message') and completion.choices[0].message:
-                return completion.choices[0].message.content
-            else:
-                return "Error: Received empty message from API."
-        else:
-            return "Error: Received incomplete response from API."
+    max_retries = 3
+    retry_delay = 2  # seconds
+    
+    for attempt in range(max_retries):
+        try:
+            completion = client.chat.completions.create(
+                extra_headers={
+                    "HTTP-Referer": "https://environmental-impact-analyzer.app",
+                    "X-Title": "Smart Environmental Impact Analyzer",
+                },
+                model="google/gemini-2.5-pro-exp-03-25:free",
+                messages=messages,
+                temperature=0.7,  # Add temperature parameter
+                max_tokens=1024,   # Limit token length to avoid timeouts
+                timeout=60         # Set timeout to 60 seconds
+            )
             
-    except Exception as e:
-        return f"Error during analysis: {str(e)}"
+            # Check if completion and choices exist before accessing
+            if completion and hasattr(completion, 'choices') and completion.choices and len(completion.choices) > 0:
+                if hasattr(completion.choices[0], 'message') and completion.choices[0].message:
+                    return completion.choices[0].message.content
+                else:
+                    if attempt < max_retries - 1:
+                        time.sleep(retry_delay)
+                        continue
+                    return "Error: Received empty message from API. Please try again."
+            else:
+                if attempt < max_retries - 1:
+                    time.sleep(retry_delay)
+                    continue
+                return "Error: Received incomplete response from API. Please try again with more specific details."
+                
+        except Exception as e:
+            if attempt < max_retries - 1:
+                time.sleep(retry_delay)
+                continue
+            return f"Error during analysis: {str(e)}. Please check your API key and try again."
+    
+    return "Failed to get a response after multiple attempts. Please try again later."
+
+def clear_output():
+    return ""
 
 # Create Gradio interface
 with gr.Blocks(title="Smart Environmental Impact Analyzer") as app:
@@ -179,15 +201,20 @@ with gr.Blocks(title="Smart Environmental Impact Analyzer") as app:
     api_key = gr.Textbox(label="OpenRouter API Key", placeholder="Enter your OpenRouter API key", type="password")
     
     with gr.Tabs():
+        # Image Analysis Tab
         with gr.TabItem("Image Analysis"):
             image_input = gr.Image(type="pil", label="Upload an image for environmental analysis")
-            image_submit = gr.Button("Analyze Image")
+            
+            with gr.Row():
+                image_submit = gr.Button("Analyze Image", variant="primary")
+                image_clear = gr.Button("Clear Response", variant="secondary")
+                
             image_output = gr.Textbox(label="Analysis Results", lines=15)
             
             image_submit.click(
                 analyze_environmental_impact,
                 inputs=[
-                    api_key,  # Directly pass the api_key component
+                    api_key,
                     gr.Textbox(value="Image Analysis", visible=False),
                     image_input,
                     gr.Textbox(value="", visible=False),
@@ -196,16 +223,27 @@ with gr.Blocks(title="Smart Environmental Impact Analyzer") as app:
                 ],
                 outputs=image_output
             )
+            
+            image_clear.click(
+                clear_output,
+                inputs=None,
+                outputs=image_output
+            )
         
+        # Geographical Assessment Tab
         with gr.TabItem("Geographical Assessment"):
             location_input = gr.Textbox(label="Location (city, region, or country)", placeholder="e.g., Paris, France")
-            location_submit = gr.Button("Analyze Location")
+            
+            with gr.Row():
+                location_submit = gr.Button("Analyze Location", variant="primary")
+                location_clear = gr.Button("Clear Response", variant="secondary")
+                
             location_output = gr.Textbox(label="Analysis Results", lines=15)
             
             location_submit.click(
                 analyze_environmental_impact,
                 inputs=[
-                    api_key,  # Directly pass the api_key component
+                    api_key,
                     gr.Textbox(value="Geographical Assessment", visible=False),
                     gr.Image(value=None, visible=False, type="pil"),
                     gr.Textbox(value="", visible=False),
@@ -215,19 +253,30 @@ with gr.Blocks(title="Smart Environmental Impact Analyzer") as app:
                 outputs=location_output
             )
             
+            location_clear.click(
+                clear_output,
+                inputs=None,
+                outputs=location_output
+            )
+            
+        # Product Assessment Tab
         with gr.TabItem("Product Assessment"):
             product_info = gr.Textbox(
                 label="Product Information", 
                 placeholder="Describe the product, materials, manufacturing process, lifecycle, etc.",
                 lines=5
             )
-            product_submit = gr.Button("Analyze Product")
+            
+            with gr.Row():
+                product_submit = gr.Button("Analyze Product", variant="primary")
+                product_clear = gr.Button("Clear Response", variant="secondary")
+                
             product_output = gr.Textbox(label="Analysis Results", lines=15)
             
             product_submit.click(
                 analyze_environmental_impact,
                 inputs=[
-                    api_key,  # Directly pass the api_key component
+                    api_key,
                     gr.Textbox(value="Product Assessment", visible=False),
                     gr.Image(value=None, visible=False, type="pil"),
                     gr.Textbox(value="", visible=False),
@@ -237,19 +286,30 @@ with gr.Blocks(title="Smart Environmental Impact Analyzer") as app:
                 outputs=product_output
             )
             
+            product_clear.click(
+                clear_output,
+                inputs=None,
+                outputs=product_output
+            )
+            
+        # Custom Query Tab
         with gr.TabItem("Custom Query"):
             custom_input = gr.Textbox(
                 label="Custom Environmental Query",
                 placeholder="Enter your environmental question or describe a scenario to analyze",
                 lines=5
             )
-            custom_submit = gr.Button("Analyze")
+            
+            with gr.Row():
+                custom_submit = gr.Button("Analyze", variant="primary")
+                custom_clear = gr.Button("Clear Response", variant="secondary")
+                
             custom_output = gr.Textbox(label="Analysis Results", lines=15)
             
             custom_submit.click(
                 analyze_environmental_impact,
                 inputs=[
-                    api_key,  # Directly pass the api_key component
+                    api_key,
                     gr.Textbox(value="Custom Query", visible=False),
                     gr.Image(value=None, visible=False, type="pil"),
                     custom_input,
@@ -258,6 +318,12 @@ with gr.Blocks(title="Smart Environmental Impact Analyzer") as app:
                 ],
                 outputs=custom_output
             )
+            
+            custom_clear.click(
+                clear_output,
+                inputs=None,
+                outputs=custom_output
+            )
     
     gr.Markdown("""
     ### Privacy Notice
@@ -269,6 +335,7 @@ with gr.Blocks(title="Smart Environmental Impact Analyzer") as app:
     2. Select the type of analysis you want to perform
     3. Provide the required information (image, location, product details, or custom query)
     4. Click the "Analyze" button for your selected tab
+    5. Use the "Clear Response" button to reset results
     """)
 
 # Launch the app