Update app.py

app.py

@@ -173,120 +173,65 @@ st.markdown("""
 # ======================================
 # Main Application Interface
 # ======================================
-st.title("🧚 Welcome to Magic Story Maker!")
-
-# File upload section
-with st.container():
-    st.subheader("Step 1: Upload Your Picture")
-    uploaded_image = st.file_uploader("Choose an image...", 
-                                    type=["png", "jpg", "jpeg"],
-                                    label_visibility="collapsed")
-
-# Initialize session state for confirmation status
-if 'confirmed' not in st.session_state:
-    st.session_state.confirmed = False
-
-# Main processing flow
-if uploaded_image is not None:
-    # Display uploaded image
-    with st.spinner("✨ Magical image processing..."):
-        image = Image.open(uploaded_image)
-        st.image(image, caption="Your Magical Image", use_column_width=True)
-
-    # Prompt selection section
-    with st.container():
-        st.subheader("Step 2: Choose Story Style")
+def main():
+    """Main application interface for Streamlit"""
+    # Page configuration
+    st.set_page_config(
+        page_title="Fantasy Adventure Generator",
+        layout="wide",
+        initial_sidebar_state="collapsed"
+    )
+    
+    # Title and description
+    st.title("🧙‍♂️ Fantasy Adventure Story Generator")
+    st.markdown("""
+    Upload an image and get:
+    - Automatic scene description
+    - AI-generated adventure story
+    - Audio version of the story
+    """)
+    
+    # File uploader
+    uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+    
+    if uploaded_file is not None:
+        # Process image
+        image = Image.open(uploaded_file).convert("RGB")
+        
+        # Layout columns
+        col1, col2 = st.columns(2)
         
-        # Create three columns for prompt buttons
-        col1, col2, col3 = st.columns(3)
         with col1:
-            if st.button("📚 Learning Story", 
-                        help="Generate educational story with life lessons",
-                        key="edu_btn"):
-                st.session_state.selected_prompt = "educational"
-                st.session_state.confirmed = False
-        with col2:
-            if st.button("🌠 Fantasy Adventure", 
-                        help="Create magical adventure story",
-                        key="fantasy_btn"):
-                st.session_state.selected_prompt = "adventure"
-                st.session_state.confirmed = False
-        with col3:
-            if st.button("🐻 Animal Friends", 
-                        help="Make story about friendly animals",
-                        key="animal_btn"):
-                st.session_state.selected_prompt = "animal"
-                st.session_state.confirmed = False
-
-    # Add confirmation button
-    with st.container():
-        st.subheader("Step 3: Confirm Selection")
-        if st.button("🔮 Start Magic Creation!", 
-                   help="Click to generate story after choosing style",
-                   type="primary"):
-            st.session_state.confirmed = True
-
-    # Only show generation when confirmed
-    if st.session_state.get('confirmed', False):
-        # Generate image caption with loading state
-        with st.spinner("🔍 Analyzing image and generating description..."):
-            image_caption = generate_image_caption(image)
-            
-            # Display caption results using CSS class
-            st.subheader("📝 Image Understanding")
-            st.markdown(f'<div class="story-container image-caption">{image_caption}</div>', 
-                      unsafe_allow_html=True)
-            st.write("")  # Add spacing
-
-        # Define prompt templates
-        PROMPT_TEMPLATES = {
-            "educational": {
-                "system": "You are a children's educator writer. Create a 100-word educational story about ",
-                "icon": "📚"
-            },
-            "adventure": {
-                "system": "You are a fantasy writer. Create a 100-word adventure story about ",
-                "icon": "🌠"
-            },
-            "animal": {
-                "system": "You are an animal themes writer. Create a 100-word animal fairy tales about ",
-                "icon": "🐻"
-            }
-        }
-
-        # Safe access with default fallback
-        selected_prompt = st.session_state.get("selected_prompt", "educational")
+            st.image(image, caption="Uploaded Image", use_column_width=True)
         
-        # Story generation section
-        with st.spinner(f"{PROMPT_TEMPLATES[selected_prompt]['icon']} Creating your story..."):
-            # Generate story content using the caption
-            selected_template = PROMPT_TEMPLATES[selected_prompt]
-            story_text = generate_story_content(
-                system_prompt=selected_template["system"],
-                user_prompt=image_caption
-            )
-            
-            # Display formatted story
-            st.subheader("✨ Your Magical Story")
-            st.markdown(f'<div class="story-container">{story_text}</div>', 
-                      unsafe_allow_html=True)
-
-        # Audio generation section
-        with st.spinner("🔮 Preparing story narration..."):
-            audio_file = generate_audio_from_story(story_text)
-            st.subheader("🎧 Listen to Your Story")
-            st.audio(data=audio_file["audio"],sample_rate=audio_file["sampling_rate"])
-    else:
-        # Show waiting message
-        st.info("ℹ️ Please select a story style and click the confirmation button to continue")
-
-# Help section
-st.markdown("---")
-st.subheader("🌟 How to Use:")
-st.info("""
-1. Upload any picture (animals, nature, or people work best!)
-2. Choose your favorite story style
-3. Click the confirmation button
-4. Wait for image analysis to complete
-5. Enjoy your personalized story and audio!
-""")
+        # Generation button
+        if st.button("✨ Generate Story & Audio"):
+            with st.spinner("Processing your request..."):
+                # Generate outputs
+                caption = generate_image_caption(image)
+                sys_prompt = "You are a fantasy writer. Create a 100-word adventure story about "
+                story = generate_story_content(sys_prompt, caption)
+                speech = generate_audio_from_story(story)
+                
+                # Display results
+                with col2:
+                    st.subheader("🔍 Scene Description")
+                    st.write(caption)
+                    
+                    st.subheader("📖 Generated Story")
+                    st.write(story)
+                    
+                    st.subheader("🔊 Audio Playback")
+                    st.audio(speech["audio"], format='audio/wav')
+                    
+                    # Download button
+                    with open(speech["audio"], "rb") as audio_file:
+                        st.download_button(
+                            label="📥 Download Audio",
+                            data=audio_file,
+                            file_name="fantasy_adventure.wav",
+                            mime="audio/wav"
+                        )
+
+if __name__ == "__main__":
+    main()