Update app.py

app.py

@@ -42,11 +42,17 @@ def generate_caption(image_base64, api_key):
         "temperature": 0,
     }
 
-    response = requests.post(api_url, json=payload, headers=headers)
-    if response.status_code == 200:
-        return response.json()
-    else:
-        return {"error": response.text}
+    try:
+        response = requests.post(api_url, json=payload, headers=headers)
+        if response.status_code == 200:
+            st.write("API response received successfully.")  # Debugging message
+            return response.json()
+        else:
+            st.error(f"API Error: {response.status_code}, {response.text}")
+            return {"error": response.text}
+    except Exception as e:
+        st.error(f"An exception occurred: {e}")
+        return {"error": str(e)}
 
 # File uploader for image
 uploaded_image = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])
@@ -66,13 +72,25 @@ if uploaded_image and API_KEY:
         st.write("Generating caption...")
         result = generate_caption(image_base64, API_KEY)
 
+        # Debug: Display the full API response
+        st.write("Full API Response:")
+        st.json(result)
+
         # Display the result
         if "error" in result:
             st.error(f"Error: {result['error']}")
         else:
-            caption = result.get("messages", [{}])[-1].get("content", [{}])[0].get("text", "No caption generated.")
-            st.subheader("Generated Caption")
-            st.write(caption)
+            # Extract caption from the response
+            try:
+                # Safely navigate the response to extract the caption
+                messages = result.get("choices", [{}])[0].get("message", {}).get("content", "")
+                if messages:
+                    st.subheader("Generated Caption")
+                    st.write(messages)
+                else:
+                    st.error("No caption generated. The response structure might have changed.")
+            except Exception as e:
+                st.error(f"Error processing the response: {e}")
 else:
     if not API_KEY:
         st.warning("Please set the `NEBIUS_API_KEY` environment variable.")