Create app.py

app.py

@@ -0,0 +1,46 @@
+import streamlit as st
+from PIL import Image
+import io
+from image_to_text import generate_initial_caption
+from caption_refiner import refine_caption
+from utils import load_image
+
+def main():
+    st.set_page_config(page_title="Image Caption Generator", layout="wide")
+    st.title("Image Caption Generator")
+
+    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)
+        
+        if st.button("Generate Caption"):
+            with st.spinner("Processing image..."):
+                # Generate initial caption
+                progress_bar = st.progress(0)
+                initial_caption = generate_initial_caption(image)
+                progress_bar.progress(50)
+                
+                # Refine caption
+                final_caption = refine_caption(initial_caption)
+                progress_bar.progress(100)
+            
+            st.success("Caption generated successfully!")
+            st.write("Generated Caption:")
+            st.write(final_caption)
+            
+            # Add copy button
+            st.text_area("Copy caption", final_caption, height=100)
+            
+            # Add download button
+            caption_bytes = final_caption.encode()
+            st.download_button(
+                label="Download Caption",
+                data=caption_bytes,
+                file_name="generated_caption.txt",
+                mime="text/plain"
+            )
+
+if __name__ == "__main__":
+    main()