Update app.py

app.py

@@ -2,29 +2,29 @@
 
 import streamlit as st
 from main import generate_and_upscale_image
+import os
 
-st.title('Text to Image Generator and Upscaler')
+def main():
 
-clipdrop_api_key = st.secrets['CLIPDROP_API_KEY']
-stability_api_key = st.secrets['STABILITY_API_KEY']
-replicate_api_token = st.secrets['REPLICATE_API_TOKEN']
+    st.title("Image Generator and Upscaler")
 
-download_path = st.text_input('Enter the download path', value='')
+    text_prompt = st.text_input("Enter your text prompt")
 
-text_prompt = st.text_input('Enter your text prompt', value='')
+    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")
 
-if st.button('Generate and Upscale Image'):
-    if text_prompt and download_path:
+    download_path = st.text_input("Enter the path where the image will be saved")
+    if not os.path.exists(download_path):
+        st.error("The provided download path does not exist. Please provide a valid path.")
+        return
+
+    if st.button("Generate and Upscale"):
         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.')
+            generate_and_upscale_image(text_prompt, clipdrop_api_key, stability_api_key, replicate_api_token, download_path)
+            st.success("The image has been successfully generated, upscaled, and saved to the specified path.")
         except Exception as e:
             st.error(f"An error occurred: {e}")
-    else:
-        st.error('Please ensure all fields are filled.')
+
+if __name__ == "__main__":
+    main()