Update app.py

app.py

@@ -80,69 +80,69 @@ def txt2img(text, style="realistic"):
         
 st.sidebar.title("Choose the task")
         
-# Streamlit web app main function
-def main():
-    with st.sidebar.expander("Audio Story"):
-
-        st.write(page_title="🎨 Image-to-Audio Story 🎧", page_icon="🖼️")
-        st.title("Turn the Image into Audio Story")
-    
-        # Allows users to upload an image file
-        uploaded_file = st.file_uploader("# 📷 Upload an image...", type=["jpg", "jpeg", "png"])
-    
-        # Parameters for LLM model (in the sidebar)
-        #st.sidebar.markdown("# LLM Inference Configuration Parameters")
-        #top_k = st.sidebar.number_input("Top-K", min_value=1, max_value=100, value=5)
-        #top_p = st.sidebar.number_input("Top-P", min_value=0.0, max_value=1.0, value=0.8)
-        #temperature = st.sidebar.number_input("Temperature", min_value=0.1, max_value=2.0, value=1.5)
-    
-        if uploaded_file is not None:
-            # Reads and saves uploaded image file
-            bytes_data = uploaded_file.read()
-            with open("uploaded_image.jpg", "wb") as file:
-                file.write(bytes_data)
+# Function for the Audio Story page
+def audio_story_page():
+    st.title("🎨 Image-to-Audio Story 🎧")
+    st.write("Turn the Image into Audio Story")
     
-            st.image(uploaded_file, caption='🖼️ Uploaded Image', use_column_width=True)
-    
-            # Initiates AI processing and story generation
-            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")
+    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)
 
-    with st.sidebar.expander("Image Generator"):
-        st.title("Stable Diffusion Image Generation")
-        st.write("This app lets you generate images using Stable Diffusion with the Euler scheduler.")
+        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:
+    if st.button("Generate Image"):
+        if prompt:
             with st.spinner("Generating image..."):
-              image = txt2img(prompt= prompt, style = image_style)
+                image = txt2img(prompt=prompt, style=image_style)
             st.image(image)
-          else:
+        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():
+    st.sidebar.title("Navigation")
+    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()