Update app.py

app.py

@@ -62,7 +62,7 @@ if uploaded_image and API_KEY:
     st.image(image, caption="Uploaded Image", use_container_width=True)
     
     if st.button("Generate Image", use_container_width=True):
-        st.write("Generating...")
+        st.write("Wait AI Is Generating Image...")
         result = generate_caption(image_base64, API_KEY)
         
         if "error" in result:
@@ -77,7 +77,14 @@ if uploaded_image and API_KEY:
                 st.image(image_url, caption="Generated Image", use_container_width=True)
                 
                 # Download button for the generated image
-                st.markdown(f"[Download Image]({image_url})", unsafe_allow_html=True)
+                st.download_button(
+    label="Download Image",
+    data=requests.get(image_url).content,  # Fetch the image data
+    file_name="generated_image.png",
+    mime="image/png",
+    use_container_width=True
+)
+
             
             except Exception as e:
                 st.error(f"Error processing the response: {e}")