Update app.py

app.py

@@ -7,18 +7,43 @@ from utils import load_image
 
 def main():
     st.set_page_config(page_title="Image Caption Generator", layout="wide")
-    st.title("Image Caption Generator")
+    
+    # Custom CSS to improve the UI
+    st.markdown("""
+    <style>
+    .big-font {
+        font-size:30px !important;
+        font-weight: bold;
+        color: #1E90FF;
+    }
+    .stButton>button {
+        background-color: #4CAF50;
+        color: white;
+        font-size: 18px;
+        padding: 10px 24px;
+        border-radius: 12px;
+        border: none;
+    }
+    .stTextInput>div>div>input {
+        background-color: #F0F8FF;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
+    st.markdown('<p class="big-font">Image Caption Generator</p>', unsafe_allow_html=True)
 
     uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
     
+    col1, col2 = st.columns(2)
+    
     if uploaded_file is not None:
         image = Image.open(uploaded_file)
-        st.image(image, caption="Uploaded Image", use_column_width=True)
+        col1.image(image, caption="Uploaded Image", use_column_width=True)
         
-        if st.button("Generate Caption"):
+        if col2.button("Generate Caption"):
             with st.spinner("Processing image..."):
                 # Generate initial caption
-                progress_bar = st.progress(0)
+                progress_bar = col2.progress(0)
                 initial_caption = generate_initial_caption(image)
                 progress_bar.progress(50)
                 
@@ -26,21 +51,23 @@ def main():
                 final_caption = refine_caption(initial_caption)
                 progress_bar.progress(100)
             
-            st.success("Caption generated successfully!")
-            st.write("Generated Caption:")
-            st.write(final_caption)
+            col2.success("Caption generated successfully!")
+            col2.markdown("### Generated Caption:")
+            col2.write(final_caption)
             
             # Add copy button
-            st.text_area("Copy caption", final_caption, height=100)
+            col2.text_area("Copy caption", final_caption, height=100)
             
             # Add download button
             caption_bytes = final_caption.encode()
-            st.download_button(
+            col2.download_button(
                 label="Download Caption",
                 data=caption_bytes,
                 file_name="generated_caption.txt",
                 mime="text/plain"
             )
+    else:
+        col1.info("Please upload an image to get started.")
 
 if __name__ == "__main__":
     main()