Update src/streamlit_app.py


src/streamlit_app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import base64
 import requests
 import json
+import io
 from PIL import Image
 
 st.set_page_config(page_title="Solar Rooftop Analyzer", layout="centered")
@@ -9,19 +10,17 @@ st.title("\U0001F31E Solar Rooftop Analysis")
 
 st.markdown("Upload a rooftop image and provide your location and budget. The system will analyze the rooftop and estimate potential solar installation ROI.")
 
-# Constants
 OPENROUTER_API_KEY = "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"
 VISION_MODEL_NAME = "opengvlab/internvl3-14b:free"
 
-# Helpers
-def analyze_image_with_openrouter(image_bytes):
-    # Convert image to JPEG bytes just in case and then base64 encode
-    img = Image.open(io.BytesIO(image_bytes)).convert("RGB")
+def analyze_image_with_openrouter(image_file):
+    # Read and convert image to JPEG bytes
+    img = Image.open(image_file).convert("RGB")
     buffer = io.BytesIO()
     img.save(buffer, format="JPEG")
     jpeg_bytes = buffer.getvalue()
-
-    encoded_image = base64.b64encode(jpeg_bytes).decode("utf-8")
+    # Base64 encode with content-type prefix
+    encoded_image = "data:image/jpeg;base64," + 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, "
@@ -38,17 +37,15 @@ def analyze_image_with_openrouter(image_bytes):
                 "role": "user",
                 "content": [
                     {"type": "text", "text": prompt},
-                    {"type": "image_base64", "image_base64": encoded_image}
+                    {"type": "image_url", "image_url": {"url": 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."}
+    return {"error": f"Failed to analyze image. Status code: {response.status_code}, Response: {response.text}"}
 
 def estimate_roi(roof_area, capacity_kw, budget):
     cost_per_kw = 65000  # INR/kW
@@ -66,32 +63,26 @@ def estimate_roi(roof_area, capacity_kw, budget):
         "within_budget": budget >= net_cost
     }
 
-# UI
 with st.form("solar_form"):
-    image = st.file_uploader("Upload Rooftop Image", type=["jpg", "jpeg", "png"])
+    uploaded_file = st.file_uploader("Upload Rooftop Image", type=["jpg", "jpeg", "png"])
     location = st.text_input("Location")
     budget = st.number_input("Budget (INR)", min_value=10000.0, step=1000.0)
     submitted = st.form_submit_button("Analyze")
-    image = Image.open(file_name)
+
 if submitted:
-    if image and location and budget:
-        st.image(image, caption="Uploaded Rooftop Image", use_column_width=True)
+    if uploaded_file and location and budget:
+        st.image(uploaded_file, 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)
-
+            ai_response = analyze_image_with_openrouter(uploaded_file)
         if "choices" in ai_response:
             try:
                 choice = ai_response["choices"][0]
-                if isinstance(choice, dict) and "message" in choice and "content" in choice["message"]:
-                    content = choice["message"]["content"]
-                else:
-                    raise ValueError("Invalid response format: 'message' or 'content' missing.")
+                # Some models may return content as a string, others as a dict
+                content = choice.get("message", {}).get("content", "")
                 content_json = json.loads(content)
                 st.success("Analysis complete!")
                 st.subheader("Rooftop Analysis")
                 st.json(content_json)
-
                 if "Roof area (sqm)" in content_json and "Estimated capacity (kW)" in content_json:
                     roi = estimate_roi(
                         roof_area=content_json["Roof area (sqm)"],