Update app.py

app.py

@@ -1,28 +1,21 @@
 import streamlit as st
 from PIL import Image
-from image_to_text import generate_initial_caption
+import io
+from image_to_text import generate_caption
 
 def main():
-    st.set_page_config(page_title="Image Caption Generator", layout="wide")
-    st.title("Image Caption Generator")
+    st.title("Image Captioning App")
 
     uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
     
     if uploaded_file is not None:
         image = Image.open(uploaded_file)
-        st.image(image, caption="Uploaded Image", use_column_width=True)
+        st.image(image, caption='Uploaded Image', use_column_width=True)
         
-        if st.button("Generate Caption"):
-            with st.spinner("Processing image..."):
-                # Generate initial caption
-                caption = generate_initial_caption(image)
-            
-            st.success("Caption generated successfully!")
+        if st.button('Generate Caption'):
+            caption = generate_caption(image)
             st.write("Generated Caption:")
             st.write(caption)
-            
-            # Add copy button
-            st.text_area("Copy caption", caption, height=100)
 
 if __name__ == "__main__":
     main()