Update app.py

app.py

@@ -1,4 +1,4 @@
-# import part
+# Import Part
 import streamlit as st
 from PIL import Image
 from gtts import gTTS
@@ -7,21 +7,20 @@ import time
 
 from transformers import pipeline
 
-
+# Generates a caption for the given image using a pre-trained model.
 def image_to_caption(image_path):
-    """Generates a caption for the given image using a pre-trained model."""
     imgtocaption = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
     caption = imgtocaption(image_path)[0]['generated_text']
     return caption
 
+# Generates a story for the caption using a pre-trained model.
 def caption_to_story(text):
-    """Generates a story for the caption using a pre-trained model."""
     captiontostory = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
     story = captiontostory(text,max_length=150,min_length=50)[0]['generated_text']
     return story
 
+# Generates an audio for the story.
 def story_to_audio(text):
-    """Generates an audio for the story."""
     audio = io.BytesIO()
     tts = gTTS(text=text, lang='en', slow=False)
     tts.write_to_fp(audio)
@@ -37,7 +36,7 @@ st.markdown("Upload an image and generate your exclusive fairy tale!")
 # File Upload
 uploaded_image = st.file_uploader("Choose a picture", type=["jpg", "jpeg", "png"], key="image_uploader")
 
-# main part
+# Main Part
 if uploaded_image is not None:
     # Display the uploaded image
     st.image(uploaded_image, caption='Uploaded Image', use_column_width=True)