Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import base64
 import requests
 import json
+from PIL import Image
+import io
 
 st.set_page_config(page_title="Solar Rooftop Analyzer", layout="centered")
 st.title("\U0001F31E Solar Rooftop Analysis")
@@ -14,7 +16,13 @@ VISION_MODEL_NAME = "opengvlab/internvl3-14b:free"
 
 # Helpers
 def analyze_image_with_openrouter(image_bytes):
-    encoded_image = base64.b64encode(image_bytes).decode("utf-8")
+    # Convert image to JPEG bytes just in case and then base64 encode
+    img = Image.open(io.BytesIO(image_bytes)).convert("RGB")
+    buffer = io.BytesIO()
+    img.save(buffer, format="JPEG")
+    jpeg_bytes = buffer.getvalue()
+
+    encoded_image = base64.b64encode(jpeg_bytes).decode("utf-8")
     prompt = (
         "Analyze the rooftop in this image. Output JSON with: [Roof area (sqm), "
         "Sunlight availability (%), Shading (Yes/No), Recommended solar panel type, "
@@ -27,18 +35,22 @@ def analyze_image_with_openrouter(image_bytes):
     payload = {
         "model": VISION_MODEL_NAME,
         "messages": [
-            {"role": "user", "content": [
-                {"type": "text", "text": prompt},
-                {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encoded_image}"}}
-            ]}
+            {
+                "role": "user",
+                "content": [
+                    {"type": "text", "text": prompt},
+                    {"type": "image_base64", "image_base64": encoded_image}
+                ]
+            }
         ]
     }
     response = requests.post("https://openrouter.ai/api/v1/chat/completions", json=payload, headers=headers)
+    st.write(f"API Response status: {response.status_code}")
+    st.write(f"API Response text: {response.text}")
     if response.status_code == 200:
         return response.json()
     return {"error": "Failed to analyze image."}
 
-
 def estimate_roi(roof_area, capacity_kw, budget):
     cost_per_kw = 65000  # INR/kW
     estimated_cost = capacity_kw * cost_per_kw
@@ -64,6 +76,7 @@ with st.form("solar_form"):
 
 if submitted:
     if image and location and budget:
+        st.image(image, caption="Uploaded Rooftop Image", use_column_width=True)
         with st.spinner("Analyzing rooftop image..."):
             image_data = image.read()
             ai_response = analyze_image_with_openrouter(image_data)