Update src/streamlit_app.py

src/streamlit_app.py

@@ -70,7 +70,11 @@ if submitted:
 
         if "choices" in ai_response:
             try:
-                content = ai_response["choices"][0]["message"]["content"]
+                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.")
                 content_json = json.loads(content)
                 st.success("Analysis complete!")
                 st.subheader("Rooftop Analysis")
@@ -90,4 +94,4 @@ if submitted:
         else:
             st.error("Failed to analyze the image. Please try again.")
     else:
-        st.warning("Please upload an image and fill all fields.")
+        st.warning("Please upload an image and fill all fields.")