Update app.py

app.py

@@ -8,10 +8,10 @@ import os
 import traceback
 
 # Set page title
-st.set_page_config(page_title="Kids Story Generator")
+st.set_page_config(page_title="Story Generator for Kids")
 
 # Title and introduction
-st.title("Kids Story Generator")
+st.title("Story Generator for Kids")
 st.write("Upload a picture and let's create a magical story!")
 
 # Initialize models with better error handling
@@ -29,8 +29,6 @@ with st.spinner("Loading models..."):
     image_to_text, story_generator, error = load_models()
     if error:
         st.error(f"Failed to load models: {error}")
-    else:
-        st.success("Models loaded successfully!")
 
 # Function to generate caption from image
 def generate_caption(image):
@@ -48,9 +46,6 @@ def generate_story(caption):
     try:
         prompt = f"Once upon a time, {caption} "
         
-        # Debug output
-        st.write(f"Prompt: {prompt}")
-        
         # Generate with increased timeout and temperature
         result = story_generator(
             prompt, 
@@ -60,9 +55,6 @@ def generate_story(caption):
             top_p=0.95
         )
         
-        # Debug output
-        st.write(f"Generation result: {result}")
-        
         if result and len(result) > 0:
             story = result[0]['generated_text']
             
@@ -78,8 +70,6 @@ def generate_story(caption):
             return story, None
         return "Story generation failed.", "No story generated"
     except Exception as e:
-        st.error(f"Error in story generation: {str(e)}")
-        st.error(traceback.format_exc())
         return "Once upon a time... (Story generation failed)", str(e)
 
 # Function to convert text to speech
@@ -107,28 +97,32 @@ if uploaded_file is not None and image_to_text is not None and story_generator i
                 # Generate caption
                 caption, caption_error = generate_caption(image)
                 if caption_error:
-                    st.warning(f"Caption generation issue: {caption_error}")
+                    st.warning(f"Caption generation issue: {caption_error}", icon="⚠️")
+                
+                # Display the caption (without debug information)
                 st.write("Image caption:", caption)
                 
                 # Generate story
                 story, story_error = generate_story(caption)
-                if story_error:
-                    st.warning(f"Story generation issue: {story_error}")
+                if story_error and not st.session_state.get("deployed", True):
+                    st.warning(f"Story generation issue: {story_error}", icon="⚠️")
+                
+                # Display the story (without debug information)
                 word_count = len(story.split())
                 st.write(f"### Your Story ({word_count} words)")
                 st.write(story)
                 
                 # Generate audio
                 audio_file, audio_error = text_to_speech(story)
-                if audio_error:
-                    st.warning(f"Audio generation issue: {audio_error}")
-                else:
+                if audio_error and not st.session_state.get("deployed", True):
+                    st.warning(f"Audio generation issue: {audio_error}", icon="⚠️")
+                elif audio_file:
                     # Display audio
                     st.write("### Listen to your story")
                     st.audio(audio_file)
     except Exception as e:
-        st.error(f"Error processing image: {str(e)}")
-        st.error(traceback.format_exc())
+        if not st.session_state.get("deployed", True):
+            st.error(f"Error processing image: {str(e)}")
 
 st.markdown("---")
-st.write("Created for ISOM5240_Assignment1")
+st.write("Created for ISOM5240 Assignment 1")