Update app.py

app.py

@@ -1,29 +1,30 @@
 # app.py
 
 import streamlit as st
-import main
+from main import generate_and_upscale_image
 
-def app():
+st.title('Text to Image Generator and Upscaler')
 
-    st.title("Image Generator and Upscaler")
+clipdrop_api_key = st.secrets['CLIPDROP_API_KEY']
+stability_api_key = st.secrets['STABILITY_API_KEY']
+replicate_api_token = st.secrets['REPLICATE_API_TOKEN']
 
-    text_prompt = st.text_input("Enter your text prompt")
-    clipdrop_api_key = st.text_input("Enter your ClipDrop API Key", type="password")
-    stability_api_key = st.text_input("Enter your Stability API Key", type="password")
-    replicate_api_token = st.text_input("Enter your Replicate API Token", type="password")
+download_path = st.text_input('Enter the download path', value='')
 
-    if st.button("Generate and Upscale Image"):
-        if text_prompt and clipdrop_api_key and stability_api_key and replicate_api_token:
-            final_img = main.generate_and_upscale_image(
-                text_prompt, clipdrop_api_key, stability_api_key, replicate_api_token
-            )
-
-            if final_img is not None:
-                st.image(final_img)
-            else:
-                st.write("An error occurred. Please try again.")
-        else:
-            st.write("Please enter all required fields.")
+text_prompt = st.text_input('Enter your text prompt', value='')
 
-if __name__ == "__main__":
-    app()
+if st.button('Generate and Upscale Image'):
+    if text_prompt and download_path:
+        try:
+            generate_and_upscale_image(
+                text_prompt,
+                clipdrop_api_key,
+                stability_api_key,
+                replicate_api_token,
+                download_path
+            )
+            st.success('Image saved successfully at the specified download path.')
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
+    else:
+        st.error('Please ensure all fields are filled.')