Update app.py

app.py

@@ -1,16 +1,30 @@
+# Import part
 import streamlit as st
 from PIL import Image
 import time
 
+# Function part
+def generate_image_caption(image_path):
+    """Generates a caption for the given image using a pre-trained model."""
+    img2caption = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+    result = img2caption(image_path)
+    return result[0]['generated_text']
+
+# text2story
+def text2story(text):
+    pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
+    story_text = pipe(text)[0]['generated_text']
+    return story_text
+
+# Main part
 # App title
-st.title("Streamlit Demo on Hugging Face")
+st.title("Assignment")
 
 # Write some text
-st.write("Welcome to a demo app showcasing basic Streamlit components!")
+st.write("Image to Story")
 
 # File uploader for image and audio
 uploaded_image = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
-uploaded_audio = st.file_uploader("Upload an audio file", type=["mp3", "wav", "ogg"])
 
 # Display image with spinner
 if uploaded_image is not None:
@@ -18,13 +32,7 @@ if uploaded_image is not None:
         time.sleep(1)  # Simulate a delay
         image = Image.open(uploaded_image)
         st.image(image, caption="Uploaded Image", use_column_width=True)
+            caption = generate_image_caption(uploaded_image)
+            st.write(f"Generated Caption: {caption}")
 
-# Play audio with spinner
-if uploaded_audio is not None:
-    with st.spinner("Loading audio..."):
-        time.sleep(1)  # Simulate a delay
-        st.audio(uploaded_audio)
-
-# Button interaction
-if st.button("Click Me"):
-    st.write("🎉 You clicked the button!")
+