Update app.py

app.py

@@ -45,7 +45,6 @@ def generate_caption(image_base64, api_key):
     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}")
@@ -72,23 +71,19 @@ 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
+        # Extract and display the generated caption
         if "error" in result:
             st.error(f"Error: {result['error']}")
         else:
-            # 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.")
+                # Extract the caption from the response
+                caption = (
+                    result.get("choices", [{}])[0]
+                    .get("message", {})
+                    .get("content", "No caption generated.")
+                )
+                st.subheader("Generated Caption")
+                st.write(caption)
             except Exception as e:
                 st.error(f"Error processing the response: {e}")
 else: