Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,7 +6,7 @@ import os
 
 
 from models.imageCaptioning import generateCaption
-# from models.storyGeneration import generateStory
+from models.storyGeneration import generateStory
 
 st.set_page_config(page_title="Image2Story Generator", layout="centered")
 st.title("Image2Story Generator")
@@ -26,15 +26,29 @@ if uploaded_file:
     image.save(image_path)
     
     # st.image(image, caption="uploaded image" , use_container_width=True)
-    # story_style = st.sidebar.selectbox("Choose Story Style" , ["William Shakespeare" , "Leo Tolstoy", "Charles Dickens" , "Haruki Murakami" , "J.K. Rowling" , "Stephen King"])  
+    story_style = st.sidebar.selectbox("Choose Story Style" , ["William Shakespeare" , "Leo Tolstoy", "Charles Dickens" , "Haruki Murakami" , "J.K. Rowling" , "Stephen King"])  
 
     
     
     with st.spinner("Generating Captions..."):
         captions = generateCaption(image_path)
         st.success(f"The image is of : {captions}")
+
+
+    story_style = st.sidebar.selectbox("Choose Story Style" , ["William Shakespeare" , "Leo Tolstoy", "Charles Dickens" , "Haruki Murakami" , "J.K. Rowling" , "Stephen King"])  
+    
+    prompt = f"""
+    You are a professional story writer with a deep understanding of surreal and introspective storytelling, inspired by {story_style}.
+    Write a short story based on the theme: {caption.strip()}.
+    The story should be in between 500 and 800 words, and must include:
+    - A clear beginning and ending.
+    - Subtle emotional depth and a dreamlike atmosphere.
+    - Elements typical of {story_style}'s writing style.
+    Do not include a title. Return only the story.
+    """
+
     
-    # with st.spinner("Generating Story..."): 
-    #     story = generateStory(captions)
-    #     st.markdown("Generated Story")
-    #     st.write(story)
+    with st.spinner("Generating Story..."): 
+        story = generateStory(captions)
+        st.markdown("Generated Story")
+        st.write(story)