Update app.py

update tampilan

app.py

@@ -1,33 +1,71 @@
 import streamlit as st
 import os
 from PIL import Image
-
 import google.generativeai as genai
+
+# Konfigurasi API Gemini
 secret_key = os.getenv("SECRET_KEY")
 genai.configure(api_key=secret_key)
 
+# Fungsi untuk menghasilkan prompt dari gambar
 def get_gemini_response(image):
-    model = genai.GenerativeModel('gemini-1.5-flash')
-    input='''You are a prompt generator.You will get an image.
+    model = genai.GenerativeModel('gemini-2.0-flash')
+    input = '''You are a prompt generator. You will get an image.
     Your work is to write a prompt such that an image generator model would create most identical picture
     as the image given to you'''
-    response = model.generate_content([input,image])  
+    response = model.generate_content([input, image])  
     return response.text
 
-st.set_page_config(page_title="Prompt generation from image")
-st.header("Application")
+# Konfigurasi halaman
+st.set_page_config(page_title="Prompt Generation from Image", layout="wide")
+
+st.title("🖼️ Prompt Generator from Image")
 
-uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png", "webp"])
+# Membuat dua kolom
+col1, col2 = st.columns(2)
 
-image=""  
+with col1:
+    st.markdown("### 📤 Upload Your Image")
+    uploaded_file = st.file_uploader(
+        "Drag and drop or click to upload an image...",
+        type=["jpg", "jpeg", "png", "webp"],
+        label_visibility="collapsed"
+    )
 
-if uploaded_file is not None:
-    image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image.", use_column_width=True)
+    if uploaded_file is not None:
+        image = Image.open(uploaded_file)
+        st.image(image, caption="Uploaded Image", use_column_width=True)
+    else:
+        image = None
 
-submit=st.button("Generate Prompt")
+with col2:
+    st.markdown("### 🎯 Generated Prompt")
+    if uploaded_file is not None:
+        if st.button("✨ Generate Prompt"):
+            with st.spinner("Generating prompt..."):
+                prompt = get_gemini_response(image)
+            st.code(prompt, language="markdown")
 
-if submit:
-    response=get_gemini_response(image)
-    st.subheader("The Generated Prompt is")
-    st.write(response)
+            # Tombol copy dengan interaksi
+            st.markdown(f"""
+                <button id="copyButton" 
+                        onclick="copyToClipboard()" 
+                        style="background-color:#4CAF50;color:white;padding:8px 12px;
+                               border:none;border-radius:4px;cursor:pointer;margin-top:10px;">
+                    📋 Copy Prompt
+                </button>
+                <script>
+                    function copyToClipboard() {{
+                        const text = `{prompt}`;
+                        navigator.clipboard.writeText(text).then(function() {{
+                            const btn = document.getElementById("copyButton");
+                            btn.innerHTML = "✅ Copied!";
+                            setTimeout(() => {{
+                                btn.innerHTML = "📋 Copy Prompt";
+                            }}, 2000);
+                        }});
+                    }}
+                </script>
+                """, unsafe_allow_html=True)
+    else:
+        st.info("Please upload an image to generate a prompt.")