Update src/streamlit_app.py

src/streamlit_app.py

@@ -26,13 +26,25 @@ 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"])  
 
     
     
+    # with st.spinner("Generating Captions..."):
+    #     captions = generateCaption(image_path)
+    #     st.success(f"The image is of : {captions}")
+
+
+
     with st.spinner("Generating Captions..."):
         captions = generateCaption(image_path)
-        st.success(f"The image is of : {captions}")
+        st.success("Caption generated successfully!")
+    
+        st.markdown(f"""
+            <div style="background-color:#f0f2f6;padding:15px 20px;border-radius:10px;margin-top:20px;border-left:5px solid #4CAF50;">
+                <h4 style="margin-bottom:10px;">🖼️ <strong>Scenario is :</strong></h4>
+                <p style="font-size:16px;color:#333;">{captions}</p>
+            </div>
+        """, unsafe_allow_html=True)
 
 
     story_style = st.sidebar.selectbox("Choose Story Style" , ["William Shakespeare" , "Leo Tolstoy", "Charles Dickens" , "Haruki Murakami" , "J.K. Rowling" , "Stephen King"])  
@@ -48,7 +60,23 @@ if uploaded_file:
     """
 
     
+    # 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)
+    
+        st.markdown("""
+            <div style="background-color:#f9f9f9;padding:15px 20px;border-radius:10px;margin-top:20px;border-left:5px solid #2196F3;">
+                <h4 style="margin-bottom:10px;">📖 <strong>Generated Story</strong></h4>
+        """, unsafe_allow_html=True)
+    
+        st.markdown(f"""
+            <div style="font-size:16px;line-height:1.6;color:#444;">
+                {story.replace('\n', '<br>')}
+            </div>
+            </div>
+        """, unsafe_allow_html=True)