Update app.py

app.py

@@ -5,6 +5,7 @@ import requests
 from PIL import Image
 import io
 import tempfile
+import base64
 
 def analyze_environmental_impact(api_key, analysis_type, image=None, text_input=None, location=None, product_info=None):
     """
@@ -30,7 +31,6 @@ def analyze_environmental_impact(api_key, analysis_type, image=None, text_input=
         
         # Convert image to base64
         with open(image_path, "rb") as img_file:
-            import base64
             image_base64 = base64.b64encode(img_file.read()).decode("utf-8")
         
         # Clean up temp file
@@ -149,14 +149,22 @@ def analyze_environmental_impact(api_key, analysis_type, image=None, text_input=
     try:
         completion = client.chat.completions.create(
             extra_headers={
-                "HTTP-Referer": "https://environmental-impact-analyzer.app",  # Replace with your actual site
+                "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
         )
         
-        return completion.choices[0].message.content
+        # 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."
+            
     except Exception as e:
         return f"Error during analysis: {str(e)}"
 
@@ -170,52 +178,57 @@ 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")
     
-    # Create hidden inputs for each tab
-    with gr.Group(visible=False) as hidden_inputs:
-        analysis_type_input = gr.Textbox(value="Image Analysis", visible=False)
-        empty_text_input = gr.Textbox(value="", visible=False)
-        empty_location_input = gr.Textbox(value="", visible=False)
-        empty_product_input = gr.Textbox(value="", visible=False)
+    # Create inputs for each analysis type
+    image_input = gr.Image(type="pil", visible=False)
+    text_input = gr.Textbox(visible=False)
+    location_input = gr.Textbox(visible=False)
+    product_input = gr.Textbox(visible=False)
     
     with gr.Tabs():
         with gr.TabItem("Image Analysis"):
-            image_input = gr.Image(type="pil", label="Upload an image for environmental analysis")
+            image_input_tab = gr.Image(type="pil", label="Upload an image for environmental analysis")
             image_submit = gr.Button("Analyze Image")
             image_output = gr.Textbox(label="Analysis Results", lines=15)
             
+            def process_image_tab():
+                return analyze_environmental_impact(
+                    api_key.value, 
+                    "Image Analysis",
+                    image=image_input_tab.value,
+                    text_input=None,
+                    location=None,
+                    product_info=None
+                )
+                
             image_submit.click(
-                analyze_environmental_impact,
-                inputs=[
-                    api_key, 
-                    gr.Textbox(value="Image Analysis", visible=False), 
-                    image_input, 
-                    empty_text_input, 
-                    empty_location_input, 
-                    empty_product_input
-                ],
+                fn=process_image_tab,
+                inputs=None,
                 outputs=image_output
             )
         
         with gr.TabItem("Geographical Assessment"):
-            location_input = gr.Textbox(label="Location (city, region, or country)", placeholder="e.g., Paris, France")
+            location_input_tab = gr.Textbox(label="Location (city, region, or country)", placeholder="e.g., Paris, France")
             location_submit = gr.Button("Analyze Location")
             location_output = gr.Textbox(label="Analysis Results", lines=15)
             
+            def process_location_tab():
+                return analyze_environmental_impact(
+                    api_key.value, 
+                    "Geographical Assessment",
+                    image=None,
+                    text_input=None,
+                    location=location_input_tab.value,
+                    product_info=None
+                )
+            
             location_submit.click(
-                analyze_environmental_impact,
-                inputs=[
-                    api_key, 
-                    gr.Textbox(value="Geographical Assessment", visible=False), 
-                    gr.Image(visible=False, type="pil"), 
-                    empty_text_input, 
-                    location_input, 
-                    empty_product_input
-                ],
+                fn=process_location_tab,
+                inputs=None,
                 outputs=location_output
             )
             
         with gr.TabItem("Product Assessment"):
-            product_info = gr.Textbox(
+            product_info_tab = gr.Textbox(
                 label="Product Information", 
                 placeholder="Describe the product, materials, manufacturing process, lifecycle, etc.",
                 lines=5
@@ -223,21 +236,24 @@ with gr.Blocks(title="Smart Environmental Impact Analyzer") as app:
             product_submit = gr.Button("Analyze Product")
             product_output = gr.Textbox(label="Analysis Results", lines=15)
             
+            def process_product_tab():
+                return analyze_environmental_impact(
+                    api_key.value, 
+                    "Product Assessment",
+                    image=None,
+                    text_input=None,
+                    location=None,
+                    product_info=product_info_tab.value
+                )
+            
             product_submit.click(
-                analyze_environmental_impact,
-                inputs=[
-                    api_key, 
-                    gr.Textbox(value="Product Assessment", visible=False), 
-                    gr.Image(visible=False, type="pil"), 
-                    empty_text_input, 
-                    empty_location_input, 
-                    product_info
-                ],
+                fn=process_product_tab,
+                inputs=None,
                 outputs=product_output
             )
             
         with gr.TabItem("Custom Query"):
-            custom_input = gr.Textbox(
+            custom_input_tab = gr.Textbox(
                 label="Custom Environmental Query",
                 placeholder="Enter your environmental question or describe a scenario to analyze",
                 lines=5
@@ -245,16 +261,19 @@ with gr.Blocks(title="Smart Environmental Impact Analyzer") as app:
             custom_submit = gr.Button("Analyze")
             custom_output = gr.Textbox(label="Analysis Results", lines=15)
             
+            def process_custom_tab():
+                return analyze_environmental_impact(
+                    api_key.value, 
+                    "Custom Query",
+                    image=None,
+                    text_input=custom_input_tab.value,
+                    location=None,
+                    product_info=None
+                )
+            
             custom_submit.click(
-                analyze_environmental_impact,
-                inputs=[
-                    api_key, 
-                    gr.Textbox(value="Custom Query", visible=False), 
-                    gr.Image(visible=False, type="pil"), 
-                    custom_input, 
-                    empty_location_input, 
-                    empty_product_input
-                ],
+                fn=process_custom_tab,
+                inputs=None,
                 outputs=custom_output
             )