Update app.py

app.py

@@ -11,6 +11,12 @@ st.set_page_config(page_title="Image Caption Generator", layout="centered")
 # API key from environment variable
 API_KEY = os.environ.get("NEBIUS_API_KEY")
 
+# UI for the app
+st.title("Image Caption Generator")
+st.write(
+    "Upload an image, and this app will generate a detailed caption for it using the Nebius AI API."
+)
+
 if not API_KEY:
     st.error("API key not found. Please set the `NEBIUS_API_KEY` environment variable.")
 
@@ -79,21 +85,15 @@ if uploaded_image and API_KEY:
                 st.subheader("Generated Caption")
                 st.write(caption)
 
-                # JavaScript for clipboard functionality (with custom HTML block)
-                st.components.v1.html(
+                # Add a copy button for the caption using JavaScript
+                st.markdown(
                     f"""
-                    <div style="text-align:center;">
-                        <textarea id="caption-textarea" style="width: 80%; height: 100px;">{caption}</textarea><br><br>
-                        <button onclick="copyToClipboard()">Copy Caption</button>
-                    </div>
+                    <button id="copy-button" onclick="navigator.clipboard.writeText('{caption}')">Copy Caption</button>
                     <script>
-                    function copyToClipboard() {{
-                        var copyText = document.getElementById("caption-textarea");
-                        copyText.select();
-                        copyText.setSelectionRange(0, 99999); // For mobile devices
-                        document.execCommand("copy");
-                        alert("Caption copied to clipboard!");
-                    }}
+                    const copyButton = document.getElementById('copy-button');
+                    copyButton.addEventListener('click', function() {{
+                        copyButton.innerText = 'Copied!';
+                    }});
                     </script>
                     """,
                     unsafe_allow_html=True,
@@ -116,7 +116,6 @@ st.markdown(
         .css-1y4ccs5 {margin: 0 auto;}
         .css-1aumxhk {display: flex; justify-content: center;}
         .css-1k6kn8p {justify-content: center; align-items: center; display: flex;}
-        .css-ffhzg6 {text-align: center;}
     </style>
     """, unsafe_allow_html=True
 )