Update app.py

app.py

@@ -9,18 +9,45 @@ API_URL = "https://api.studio.nebius.ai/v1/chat/completions"
 
 # Streamlit app configuration
 st.set_page_config(page_title="Image to Prompt Converter", layout="centered", page_icon="🖼️")
-st.markdown("<style>body { background-color: #1E1E1E; color: #FFFFFF; }</style>", unsafe_allow_html=True)
+
+# Apply custom styles
+st.markdown(
+    """
+    <style>
+    body {
+        background: linear-gradient(135deg, #1e3c72, #2a5298);
+        color: #FFFFFF;
+        font-family: 'Arial', sans-serif;
+    }
+    .stApp {
+        align-items: center;
+        justify-content: center;
+    }
+    .css-1offfwp {
+        margin: auto !important;
+    }
+    h1, h2, h3 {
+        text-align: center;
+    }
+    .uploaded-image {
+        border-radius: 10px;
+        margin-bottom: 20px;
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
 
 # App title
 st.title("Image to Prompt Converter")
-st.markdown("Upload an image and generate a detailed prompt.")
+st.markdown("**Upload an image and generate a detailed prompt.**")
 
 # Image upload
 uploaded_file = st.file_uploader("Upload an image", type=["jpg", "png", "jpeg"])
 
 if uploaded_file is not None:
     # Display the uploaded image
-    st.image(uploaded_file, caption="Uploaded Image", use_column_width=True)
+    st.image(uploaded_file, caption="Uploaded Image", use_container_width=True, class_="uploaded-image")
     
     # Generate button
     if st.button("Generate Prompt"):
@@ -51,8 +78,6 @@ if uploaded_file is not None:
             st.text_area("", generated_prompt, height=200)
             
             # Copy button
-            if st.button("Copy Prompt"):
-                st.write("Prompt copied to clipboard!")
-                st.write(generated_prompt)
+            st.button("Copy Prompt", on_click=lambda: st.session_state.update({"clipboard": generated_prompt}))
         else:
             st.error(f"Failed to generate prompt: {response.status_code} - {response.text}")