Update app.py

app.py

@@ -58,21 +58,25 @@ if uploaded_image and API_KEY:
     image.save(buffered, format="PNG")
     image_base64 = base64.b64encode(buffered.getvalue()).decode()
 
-    # Generate caption
+    # Show the uploaded image
     st.image(image, caption="Uploaded Image", use_column_width=True)
-    st.write("Generating caption...")
-    result = generate_caption(image_base64, API_KEY)
 
-    # 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)
-elif not API_KEY:
-    st.warning("Please set the `NEBIUS_API_KEY` environment variable.")
+    # Add a button to trigger caption generation
+    if st.button("Generate Caption"):
+        st.write("Generating caption...")
+        result = generate_caption(image_base64, API_KEY)
+
+        # 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)
 else:
-    st.info("Please upload an image.")
+    if not API_KEY:
+        st.warning("Please set the `NEBIUS_API_KEY` environment variable.")
+    else:
+        st.info("Please upload an image.")
 
 st.sidebar.write("Built with ❤️ by OpenAI GPT-4")