Update app.py

app.py

@@ -4,23 +4,24 @@ import requests
 from groq import Groq
 
 # Initialize Groq client
-visionkey = os.getenv("GroqVision")
+visionkey = os.getenv("GroqVision")  # Replace with your actual Groq API key if not set in the environment
 client = Groq(api_key=visionkey)
 
 # Your imgbb API key
-imagekey = os.getenv("ImageAPI")
+imagekey = os.getenv("ImageAPI")  # Replace with your actual imgbb API key if not set in the environment
 IMGBB_API_KEY = imagekey
 
 def upload_image_to_imgbb(image_path):
     """Uploads an image to imgbb and returns the URL."""
-    url = f"https://api.imgbb.com/1/upload?key={IMGBB_API_KEY}"
-    with open(image_path, "rb") as image_file:
-        payload = {"image": image_file.read()}
-        response = requests.post(url, files=payload)
-        if response.status_code == 200:
-            return response.json()["data"]["url"]
-        else:
-            raise ValueError(f"Image upload failed: {response.json()}")
+    try:
+        url = f"https://api.imgbb.com/1/upload?key={IMGBB_API_KEY}"
+        with open(image_path, "rb") as image_file:
+            files = {"image": image_file}  # Corrected: Use 'files' instead of 'data'
+            response = requests.post(url, files=files, timeout=30)  # Added timeout for better control
+            response.raise_for_status()  # Raises an error for HTTP codes 4xx/5xx
+        return response.json()["data"]["url"]
+    except requests.exceptions.RequestException as e:
+        raise ValueError(f"Image upload failed: {str(e)}")
 
 def analyze_image(image, instruction):
     """Analyzes the image using Groq's Llama 3.2 based on the instruction provided."""
@@ -35,13 +36,10 @@ def analyze_image(image, instruction):
         # Call the Groq API to analyze the image
         completion = client.chat.completions.create(
             model="llama-3.2-90b-vision-preview",
-            messages=[{
-                "role": "user",
-                "content": [
-                    {"type": "text", "text": instruction},
-                    {"type": "image_url", "image_url": {"url": image_url}}
-                ]
-            }],
+            messages=[
+                {"role": "user", "content": instruction},
+                {"role": "user", "content": {"type": "image_url", "image_url": {"url": image_url}}},
+            ],
             temperature=1,
             max_tokens=1024,
             top_p=1,
@@ -50,7 +48,7 @@ def analyze_image(image, instruction):
         )
 
         # Extract and return the response
-        analysis = completion.choices[0].message.content  # Fix here to access content directly
+        analysis = completion.choices[0].message.content
         return analysis
     except Exception as e:
         return f"Error: {str(e)}"
@@ -64,9 +62,9 @@ iface = gr.Interface(
     ],
     outputs="text",
     title="Deep Image Analysis using LLM",
-    description="Upload an image and provide instructions to analyze the image using Llama 3.2 90B Vision. You can upload and analyse as many pictures as possible with no restrictions. However, it's one at a time.",
+    description="Upload an image and provide instructions to analyze the image using Llama 3.2 90B Vision. You can upload and analyze as many pictures as possible with no restrictions. However, it's one at a time.",
     live=False,
 )
 
 # Launch the app
-iface.launch()
+iface.launch()