Update app.py

app.py

@@ -13,9 +13,15 @@ GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 client = Groq(api_key=GROQ_API_KEY)
 
 def encode_image(image):
-    buffered = io.BytesIO()
-    image.save(buffered, format="PNG")
-    return base64.b64encode(buffered.getvalue()).decode('utf-8')
+    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")
 
 def analyze_construction_image(image, follow_up_question=""):
     if image is None:
@@ -49,7 +55,7 @@ def analyze_construction_image(image, follow_up_question=""):
             })
 
         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,
@@ -87,7 +93,7 @@ iface = gr.Interface(
         ["example_image1.jpg", "What safety concerns do you see?"],
         ["example_image2.jpg", "Is there any visible structural damage?"]
     ],
-    cache_examples=True,
+    cache_examples=False,  # Disable caching to avoid file moving issues
     theme="default"
 )