Update app.py

app.py

@@ -80,55 +80,68 @@ def txt2img(text, style="realistic"):
         
 st.sidebar.title("Choose the task")
         
-# Streamlit web app main function
-def main():
-    st.title("Welcome to your Creative Canvas!")
-    st.write("Use the tools in the sidebar to create audio stories and unique images.")
-    
-    with st.expander("Audio Story"):
-        st.write(page_title="🎨 Image-to-Audio Story 🎧", page_icon="🖼️")
-        st.title("Turn the Image into Audio Story")
-        
-        uploaded_file = st.file_uploader("# 📷 Upload an image...", type=["jpg", "jpeg", "png"])
-        
-        if uploaded_file is not None:
-            bytes_data = uploaded_file.read()
-            with open("uploaded_image.jpg", "wb") as file:
-                file.write(bytes_data)
-            
-            st.image(uploaded_file, caption='🖼️ Uploaded Image', use_column_width=True)
-            
-            with st.spinner("## 🤖 AI is at Work! "):
-                scenario = img2txt("uploaded_image.jpg")  # Extracts text from the image
-                story = generate_story(scenario, llm)  # Generates a story based on the image text, LLM params
-                txt2speech(story)  # Converts the story to audio
-                
-                st.markdown("---")
-                st.markdown("## 📜 Image Caption")
-                st.write(scenario)
-                
-                st.markdown("---")
-                st.markdown("## 📖 Story")
-                st.write(story)
-                
-                st.markdown("---")
-                st.markdown("## 🎧 Audio Story")
-                st.audio("audio_story.mp3")
+# Function for the Audio Story page
+def audio_story_page():
+    st.title("🎨 Image-to-Audio Story 🎧")
+    st.write("Turn the Image into Audio Story")
     
-    with st.expander("Image Generator"):
-        st.title("Stable Diffusion Image Generation")
-        st.write("This app lets you generate images using Stable Diffusion with the Euler scheduler.")
+    uploaded_file = st.file_uploader("Upload an image...", type=["jpg", "jpeg", "png"])
+
+    if uploaded_file is not None:
+        bytes_data = uploaded_file.read()
+        with open("uploaded_image.jpg", "wb") as file:
+            file.write(bytes_data)
+
+        st.image(uploaded_file, caption='🖼️ Uploaded Image', use_column_width=True)
+
+        with st.spinner("AI is at Work!"):
+            scenario = img2txt("uploaded_image.jpg")
+            story = generate_story(scenario, llm)
+            txt2speech(story)
+
+            st.markdown("---")
+            st.markdown("## 📜 Image Caption")
+            st.write(scenario)
+
+            st.markdown("---")
+            st.markdown("## 📖 Story")
+            st.write(story)
+
+            st.markdown("---")
+            st.markdown("## 🎧 Audio Story")
+            st.audio("audio_story.mp3")
+
+# Function for the Image Generator page
+def image_generator_page():
+    st.title("Stable Diffusion Image Generation")
+    st.write("This app lets you generate images using Stable Diffusion with the Euler scheduler.")
         
-        prompt = st.text_input("Enter your prompt:")
-        image_style = st.selectbox("Style Selection", ["realistic", "cartoon", "watercolor"])
+    prompt = st.text_input("Enter your prompt:")
+    image_style = st.selectbox("Style Selection", ["realistic", "cartoon", "watercolor"])
         
-        if st.button("Generate Image"):
-            if prompt:
-                with st.spinner("Generating image..."):
-                    image = txt2img(prompt=prompt, style=image_style)
-                st.image(image)
-            else:
-                st.error("Please enter a prompt.")
+    if st.button("Generate Image"):
+        if prompt:
+            with st.spinner("Generating image..."):
+                image = txt2img(prompt=prompt, style=image_style)
+            st.image(image)
+        else:
+            st.error("Please enter a prompt.")
+
+# Function for the Home page
+def home_page():
+    st.title("Welcome to your Creative Canvas!")
+    st.write("Use the tools in the sidebar to create audio stories and unique images.")
+
+# Streamlit web app main function
+def main():
+    selection = st.sidebar.radio("Go to", ["Home", "Audio Story", "Image Generator"])
+
+    if selection == "Home":
+        home_page()
+    elif selection == "Audio Story":
+        audio_story_page()
+    elif selection == "Image Generator":
+        image_generator_page()
 
 if __name__ == '__main__':
     main()