Update app.py

app.py

@@ -35,7 +35,7 @@ if st.button("Generate Image"):
         with st.spinner("Generating image..."):
             result = generate_image()
             
-            image_url = result.get("image_url")
+            image_url = result.get("message", {}).get("output_png")
             if image_url:
                 st.image(image_url, caption="Generated Image", use_container_width=True)