Update app.py

app.py

@@ -5,7 +5,6 @@ import gradio as gr
 from google import genai
 from google.genai import types
 
-# Helpers
 def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
         f.write(data)
@@ -15,6 +14,7 @@ def generate_edit(prompt, pil_image, api_key, model="gemini-2.0-flash-exp"):
     with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp_img:
         image_path = tmp_img.name
         pil_image.save(image_path)
+
     files = [client.files.upload(file=image_path)]
     contents = [
         types.Content(
@@ -33,8 +33,10 @@ def generate_edit(prompt, pil_image, api_key, model="gemini-2.0-flash-exp"):
         response_modalities=["image", "text"],
         response_mime_type="text/plain",
     )
+
     text_response = ""
     image_out_path = None
+
     with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp_out:
         out_path = tmp_out.name
         for chunk in client.models.generate_content_stream(
@@ -51,7 +53,12 @@ def generate_edit(prompt, pil_image, api_key, model="gemini-2.0-flash-exp"):
                 break
             else:
                 text_response += chunk.text + "\n"
-    del files
+
+    try:
+        del files
+    except Exception:
+        pass
+
     return image_out_path, text_response
 
 def process_image_and_prompt(pil_image, prompt, api_key, progress_callback=None):
@@ -78,37 +85,42 @@ def process_image_and_prompt(pil_image, prompt, api_key, progress_callback=None)
 def reset_inputs(api_key_value=None):
     return None, "", api_key_value or "", ""
 
+# CSS: two-column layout with responsive behavior; status bar kept separate
 css_style = """
-:root { --bg:#0f111a; --panel:#151a24; --text:#e9eefc; --muted:#9fb3c8; --accent:#6a8efd; }
+:root { --bg: #0f111a; --panel: #1b1e28; --text: #e9eefc; --muted: #9fb3c8; --accent: #6a8efd; }
 body, .app-container { background: var(--bg); color: var(--text); }
-.header-block { width: 100%; display: flex; justify-content: center; padding: 12px 0; }
-.header-gradient { width: 100%; padding: 22px 0; border-radius: 12px; background: linear-gradient(90deg, #6a8efd, #44abc7); box-shadow: 0 2px 12px rgb(50 50 70 / 12%); text-align: center; }
+.header-block { width: 100%; display: flex; justify-content: center; padding: 8px 0; }
+.header-gradient { width: 100%; padding: 20px 0; border-radius: 12px; background: linear-gradient(90deg, #6a8efd, #44abc7); text-align: center; }
 .header-title { margin: 0; font-size: 2.6rem; font-weight: 900; color: #fff; text-shadow: 0 2px 8px rgba(0,0,0,.25); }
 .header-subtitle { margin-top: 6px; font-size: 1.05rem; color: #e8f0ff; }
 
-.main { display: flex; gap: 20px; align-items: stretch; padding: 0 16px; }
-.sidebar { width: 320px; background: #1b1f2b; padding: 16px; border-radius: 12px; }
-.sidebar h2 { color: #89b4ff; font-size: 1rem; margin: 8px 0; }
-.sidebar ul { margin: 0; padding-left: 20px; color: #d6e3ff; line-height: 1.9; }
-.main-panel { flex: 1; display: flex; flex-direction: column; gap: 12px; }
+.main { display: flex; gap: 20px; align-items: stretch; padding: 0 12px; }
+.sidebar { width: 320px; background: #1a1e2a; padding: 14px; border-radius: 12px; min-height: 360px; box-shadow: 0 2px 10px rgb(0 0 0 / 0.25); }
+.sidebar h2 { color: #8ab4ff; font-size: 1rem; margin: 6px 0; }
+.sidebar ul { margin: 0; padding-left: 18px; color: #d6e2ff; line-height: 1.9; }
+.sidebar a { color: #97b7ff; text-decoration: none; }
+.sidebar a:hover { text-decoration: underline; }
 
-.layout-row { display: flex; gap: 16px; align-items: flex-start; }
-.left, .right { flex: 1; min-width: 0; }
+.main-panel { flex: 1; display: flex; flex-direction: column; gap: 12px; }
 
-.section-header { font-weight: 800; font-size: 1.05rem; color: #cbd5e1; margin: 0 0 6px 0; }
+.section-header { font-weight: 800; font-size: 1.04rem; color: #cbd5e1; margin: 6px 0; }
 
+.layout-two-col {
+  display: grid;
+  grid-template-columns: 1fr 1fr;
+  gap: 14px;
+}
+@media (max-width: 1100px) {
+  .layout-two-col { grid-template-columns: 1fr; }
+}
 .input-area, .output-area { background: #202533; border-radius: 12px; padding: 12px; }
-#status-text { height: 1.6em; line-height: 1.6em; white-space: nowrap; overflow: hidden; text-overflow: ellipsis; padding: 0 8px; border: none; background: transparent; color: #cbd5e1; font-weight: 700; }
-#output-viewport { display: flex; justify-content: center; align-items: center; min-height: 260px; }
-#output-image { width: 100%; display: flex; justify-content: center; align-items: center; }
+#status-text { height: 1.4em; line-height: 1.4em; white-space: nowrap; overflow: hidden; text-overflow: ellipsis; padding: 0 8px; border: none; background: transparent; font-weight: 700; color: #cbd5e1; }
+#output-image { display: flex; justify-content: center; align-items: center; padding: 6px; }
 #output-image img { max-width: 100%; max-height: 420px; object-fit: contain; border-radius: 12px; background: #23252b; }
 
-@media (max-width: 1100px) {
-  .main { flex-direction: column; }
-  .sidebar { display: none; }
-}
 """
 
+# Layout
 with gr.Blocks(css=css_style) as app:
     gr.HTML("""
     <div class='header-block'>
@@ -118,6 +130,7 @@ with gr.Blocks(css=css_style) as app:
       </div>
     </div>
     """)
+
     with gr.Row():
         with gr.Column(scale=3, elem_classes="sidebar"):
             gr.Markdown(
@@ -134,17 +147,15 @@ with gr.Blocks(css=css_style) as app:
                 """
             )
         with gr.Column(scale=9, elem_classes="main-panel"):
-            # Step 1: Upload on left; Step 3: Output on right
-            # We’ll implement side-by-side using a two-column sub-layout
-            with gr.Row(class_name="layout-row"):
-                with gr.Column(class_name="left"):
+            # Step 1 & Step 3 side-by-side area
+            with gr.Row():
+                with gr.Column():
                     gr.Markdown("<div class='section-header'>Step 1: Upload Image</div>")
                     image_input = gr.Image(type="pil", label=None, image_mode="RGBA")
-                with gr.Column(class_name="right"):
+                with gr.Column():
                     gr.Markdown("<div class='section-header'>Step 3: Image Output</div>")
                     output_image = gr.Image(label=None, show_label=False, type="pil")
-
-            # Step 2 area (prompt + API)
+            # Step 2 inputs
             gr.Markdown("<div class='section-header'>Step 2: Enter Editing Prompt</div>")
             prompt_input = gr.Textbox(label="Edit Prompt", placeholder="Describe how to edit the image", lines=2)
             api_key_input = gr.Textbox(label="Gemini API Key (required)", placeholder="Enter your Gemini API key here", type="password")
@@ -154,10 +165,8 @@ with gr.Blocks(css=css_style) as app:
                 reset_btn = gr.Button("Reset Inputs")
 
             with gr.Row():
-                # Status bar (persistent)
                 status_bar = gr.Textbox(label="Status", interactive=False, lines=1, elem_id="status-text")
 
-            # Wire callbacks
             def on_submit(pil_img, prompt, key, progress=None):
                 if not key or key.strip() == "":
                     raise gr.Error("Gemini API Key is required!")