Update app.py

app.py

@@ -46,15 +46,23 @@ if generate_prompt_button:
 with st.form(key='image_form'):
     generate_image_button = st.form_submit_button(label='Generate Image')
 
+
+
 if generate_image_button and st.session_state['dynamic_prompt']:
-    # Here, integrate with your custom model to generate the image
-    # For now, using a placeholder URL
-    st.session_state['image_url'] = "url_of_your_generated_image"
-    st.session_state['image_generated'] = True
+    # Integrate with your custom model to generate the image
+    # Assuming 'my_custom_model_generate_image' is a function that returns the image URL
+    st.session_state['image_url'] = my_custom_model_generate_image(st.session_state['dynamic_prompt'])
+    st.session_state['image_generated'] = st.session_state['image_url'] is not None
 
-    # Display the generated image
+    # Display the generated image if the URL is valid
     if st.session_state['image_url']:
-        st.image(st.session_state['image_url'], caption="Generated Image", use_column_width=True)
+        try:
+            st.image(st.session_state['image_url'], caption="Generated Image", use_column_width=True)
+        except Exception as e:
+            st.error(f"Failed to load image from URL: {e}")
+
+
+
 
 # Download button for the generated image
 if st.session_state['image_generated']: