Update app.py

app.py

@@ -5,29 +5,43 @@ from PIL import Image
 import io
 import json
 from groq import Groq
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
 
 # Load environment variables
 GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
+if not GROQ_API_KEY:
+    logger.error("GROQ_API_KEY is not set in environment variables")
+    raise ValueError("GROQ_API_KEY is not set")
 
 # Initialize Groq client
 client = Groq(api_key=GROQ_API_KEY)
 
 def encode_image(image):
-    if isinstance(image, str):  # If image is a file path
-        with open(image, "rb") as image_file:
-            return base64.b64encode(image_file.read()).decode('utf-8')
-    elif isinstance(image, Image.Image):  # If image is a PIL Image
-        buffered = io.BytesIO()
-        image.save(buffered, format="PNG")
-        return base64.b64encode(buffered.getvalue()).decode('utf-8')
-    else:
-        raise ValueError("Unsupported image type")
+    try:
+        if isinstance(image, str):  # If image is a file path
+            with open(image, "rb") as image_file:
+                return base64.b64encode(image_file.read()).decode('utf-8')
+        elif isinstance(image, Image.Image):  # If image is a PIL Image
+            buffered = io.BytesIO()
+            image.save(buffered, format="PNG")
+            return base64.b64encode(buffered.getvalue()).decode('utf-8')
+        else:
+            raise ValueError(f"Unsupported image type: {type(image)}")
+    except Exception as e:
+        logger.error(f"Error encoding image: {str(e)}")
+        raise
 
 def analyze_construction_image(image, follow_up_question=""):
     if image is None:
+        logger.warning("No image provided")
         return "Error: No image uploaded", "", ""
 
     try:
+        logger.info("Starting image analysis")
         image_data_url = f"data:image/png;base64,{encode_image(image)}"
 
         messages = [
@@ -36,7 +50,7 @@ def analyze_construction_image(image, follow_up_question=""):
                 "content": [
                     {
                         "type": "text",
-                        "text": "Analyze this construction site image. Identify any issues or snags, categorize them, provide a detailed description, and suggest steps to resolve them. Output the result in JSON format."
+                        "text": "Analyze this construction site image. Identify any issues or snags, categorize them, provide a detailed description, and suggest steps to resolve them. Format your response as a JSON object with keys 'snag_category', 'snag_description', and 'desnag_steps' (as an array)."
                     },
                     {
                         "type": "image_url",
@@ -54,8 +68,9 @@ def analyze_construction_image(image, follow_up_question=""):
                 "content": follow_up_question
             })
 
+        logger.info("Sending request to Groq API")
         completion = client.chat.completions.create(
-            model="llama-3.2-11b-vision-preview",
+            model="llama-3.2-90b-vision-preview",
             messages=messages,
             temperature=0.7,
             max_tokens=1000,
@@ -65,14 +80,25 @@ def analyze_construction_image(image, follow_up_question=""):
             stop=None
         )
 
-        result = json.loads(completion.choices[0].message.content)
+        logger.info("Received response from Groq API")
+        result = completion.choices[0].message.content
+        logger.debug(f"Raw API response: {result}")
+
+        # Try to parse the result as JSON
+        try:
+            parsed_result = json.loads(result)
+        except json.JSONDecodeError:
+            logger.error("Failed to parse API response as JSON")
+            return "Error: Invalid response format", "", ""
 
-        snag_category = result.get('snag_category', 'N/A')
-        snag_description = result.get('snag_description', 'N/A')
-        desnag_steps = '\n'.join(result.get('desnag_steps', ['N/A']))
+        snag_category = parsed_result.get('snag_category', 'N/A')
+        snag_description = parsed_result.get('snag_description', 'N/A')
+        desnag_steps = '\n'.join(parsed_result.get('desnag_steps', ['N/A']))
 
+        logger.info("Analysis completed successfully")
         return snag_category, snag_description, desnag_steps
     except Exception as e:
+        logger.error(f"Error during image analysis: {str(e)}")
         return f"Error: {str(e)}", "", ""
 
 # Create the Gradio interface
@@ -93,10 +119,10 @@ iface = gr.Interface(
         ["example_image1.jpg", "What safety concerns do you see?"],
         ["example_image2.jpg", "Is there any visible structural damage?"]
     ],
-    cache_examples=False,  # Disable caching to avoid file moving issues
+    cache_examples=False,
     theme="default"
 )
 
 # Launch the app
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(debug=True)