Update app.py

app.py

@@ -26,7 +26,8 @@ if st.button("Extract Head"):
         with st.spinner("Extracting head..."):
             result = extract_head(uploaded_file)
             
-            head_url = result.get("head_image_url")
+            # Extract image URL from JSON response
+            head_url = result.get("message", {}).get("head_image_url")
             if head_url:
                 st.image(head_url, caption="Extracted Head", use_container_width=True)
                 
@@ -41,4 +42,4 @@ if st.button("Extract Head"):
                 st.error("Failed to extract head. No image URL found in response.")
                 st.write("Response Data:", result)  # Debugging: Show response data
     else:
-        st.warning("Please upload an image.")
+        st.warning("Please upload an image.")