Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
-from openai import OpenAI
 import base64
 import io
+import requests
 
 def solve_stem_problem(api_key, image, subject):
-    if api_key.strip() == "":
+    if not api_key.strip():
         return "❌ Error: API key is required."
 
     detectives = {
@@ -16,28 +16,25 @@ def solve_stem_problem(api_key, image, subject):
     detective = detectives.get(subject, "Algebra Ace")
 
     try:
-        # Convert PIL Image to base64
+        # Convert image to base64 string
         buffered = io.BytesIO()
-        image_format = "PNG"
-        image.save(buffered, format=image_format)
+        image.save(buffered, format="PNG")
         encoded_image = base64.b64encode(buffered.getvalue()).decode('utf-8')
         image_url_data = f"data:image/png;base64,{encoded_image}"
     except Exception as e:
         return f"❌ Error encoding image: {str(e)}"
 
     try:
-        client = OpenAI(
-            base_url="https://openrouter.ai/api/v1",
-            api_key=api_key,
-        )
+        headers = {
+            "Authorization": f"Bearer {api_key}",
+            "Content-Type": "application/json",
+            "HTTP-Referer": "https://stem-sleuth.example.com",
+            "X-Title": "STEM Sleuth"
+        }
 
-        completion = client.chat.completions.create(
-            extra_headers={
-                "HTTP-Referer": "https://stem-sleuth.example.com",
-                "X-Title": "STEM Sleuth",
-            },
-            model="google/gemini-2.0-flash-exp:free",
-            messages=[
+        payload = {
+            "model": "google/gemini-2.0-flash-exp:free",
+            "messages": [
                 {
                     "role": "user",
                     "content": [
@@ -47,17 +44,27 @@ def solve_stem_problem(api_key, image, subject):
                         },
                         {
                             "type": "image_url",
-                            "image_url": {"url": image_url_data}
+                            "image_url": {
+                                "url": image_url_data
+                            }
                         }
                     ]
                 }
             ]
-        )
+        }
 
-        if completion.choices and len(completion.choices) > 0:
-            return completion.choices[0].message.content
-        else:
-            return "⚠️ Could not retrieve a solution from the API."
+        response = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload)
+
+        if response.status_code != 200:
+            return f"❌ API Error {response.status_code}: {response.text}"
+
+        completion = response.json()
+        message = completion.get("choices", [{}])[0].get("message", {}).get("content")
+
+        if not message:
+            return "⚠️ No response content received. Try another image or check the model output."
+
+        return message
 
     except Exception as e:
         return f"❌ API call failed: {str(e)}"
@@ -75,4 +82,4 @@ iface = gr.Interface(
     description="Upload a math, physics, chemistry, or coding problem image and solve it with a detective twist using OpenRouter Gemini model."
 )
 
-iface.launch()
+iface.launch()