Update app.py

app.py

@@ -4,8 +4,9 @@ import os
 import base64
 from PIL import Image
 from io import BytesIO  # Correctly import BytesIO from the io module
+import pyperclip  # To copy the caption to clipboard
 
-# Set page title
+# Set page title and layout
 st.set_page_config(page_title="Image Caption Generator", layout="centered")
 
 # API key from environment variable
@@ -64,10 +65,10 @@ if uploaded_image and API_KEY:
     image_base64 = base64.b64encode(buffered.getvalue()).decode()
 
     # Show the uploaded image
-    st.image(image, caption="Uploaded Image", use_column_width=True)
+    st.image(image, caption="Uploaded Image", use_container_width=True)
 
     # Add a button to trigger caption generation
-    if st.button("Generate Caption"):
+    if st.button("Generate Caption", use_container_width=True):
         st.write("Generating caption...")
         result = generate_caption(image_base64, API_KEY)
 
@@ -84,6 +85,12 @@ if uploaded_image and API_KEY:
                 )
                 st.subheader("Generated Caption")
                 st.write(caption)
+
+                # Add a copy button for the caption
+                if st.button("Copy Caption", key="copy_button"):
+                    pyperclip.copy(caption)  # Copy the caption to clipboard
+                    st.success("Caption copied to clipboard!")
+
             except Exception as e:
                 st.error(f"Error processing the response: {e}")
 else:
@@ -92,4 +99,15 @@ else:
     else:
         st.info("Please upload an image.")
 
-st.sidebar.write("Built with ❤️ by OpenAI GPT-4")
+# Remove the sidebar and center the content
+st.markdown(
+    """
+    <style>
+        .css-1d391kg {text-align: center;}
+        .css-1v3fvcr {justify-content: center; display: flex;}
+        .css-1y4ccs5 {margin: 0 auto;}
+        .css-1aumxhk {display: flex; justify-content: center;}
+        .css-1k6kn8p {justify-content: center; align-items: center; display: flex;}
+    </style>
+    """, unsafe_allow_html=True
+)