Update app.py

app.py

@@ -1,14 +1,13 @@
 import os
 import gradio as gr
-import json
 import traceback
+from PIL import Image, ImageEnhance, ImageFilter
 import io
 import base64
-from PIL import Image, ImageEnhance, ImageFilter
+import google.generativeai as genai
 
 # --- Environment Configuration ---
 GEMINI_KEY = os.environ.get("GEMINI_KEY", "")
-DEFAULT_PORT = int(os.environ.get("PORT", 7860))
 
 # --- Style Template Optimization ---
 BASE_TEMPLATE = """Describe this design as a concise Flux 1.1 Pro prompt focusing on:
@@ -28,7 +27,6 @@ STYLE_INSTRUCTIONS = {
 # --- Flux Configuration ---
 FLUX_SPECS = {
     "aspect_ratios": ["1:1", "16:9", "4:3", "9:16"],
-    "formats": ["SVG", "PNG", "PDF"],
     "color_modes": ["B&W", "CMYK", "RGB"],
     "dpi_options": [72, 150, 300, 600]
 }
@@ -50,24 +48,23 @@ def generate_prompt(image, api_key, style, creativity, neg_prompt, aspect, color
     try:
         # Step 1: Input Validation
         if not image:
-            return "⚠️ Please upload an image.", None, None, "⚠️ No image uploaded."
+            return "", "⚠️ Please upload an image."
 
         api_key = api_key or GEMINI_KEY
         if not api_key:
-            return "🔑 API key required - set in env (GEMINI_KEY) or input field.", None, None, "🔑 Missing Gemini API key."
+            return "", "🔑 API key required - set in env (GEMINI_KEY) or input field."
 
         # Step 2: Gemini Setup
         try:
-            import google.generativeai as genai
             genai.configure(api_key=api_key)
             model = genai.GenerativeModel("gemini-1.5-pro")
         except ImportError:
-            return "", None, None, "🚫 Failed to import google.generativeai. Install with: pip install google-generativeai"
+            return "", "🚫 Failed to import google.generativeai. Install with: pip install google-generativeai"
         except Exception as e:
             if "authentication" in str(e).lower():
-                return "", None, None, "🔐 Invalid API key or authentication error."
+                return "", "🔐 Invalid API key or authentication error."
             else:
-                return "", None, None, f"⚙️ API initialization error: {str(e)}"
+                return "", f"⚙️ API initialization error: {str(e)}"
 
         # Step 3: Preprocess Image
         try:
@@ -76,14 +73,14 @@ def generate_prompt(image, api_key, style, creativity, neg_prompt, aspect, color
             img.save(img_bytes, format="PNG")
             img_b64 = base64.b64encode(img_bytes.getvalue()).decode()
         except Exception as e:
-            return "", None, None, f"🖼️ Image preparation failed: {str(e)}"
+            return "", f"🖼️ Image preparation failed: {str(e)}"
 
         # Step 4: Build Instruction Prompt
         try:
             instruction = f"{STYLE_INSTRUCTIONS[style]}\nAVOID: {neg_prompt}\n"
             instruction += f"ASPECT: {aspect}, COLORS: {color_mode}, DPI: {dpi}\n"
         except KeyError:
-            return "", None, None, "🛠️ Invalid style selected. Please choose from available options."
+            return "", "🛠️ Invalid style selected. Please choose from available options."
 
         # Step 5: Call Gemini API
         try:
@@ -93,34 +90,20 @@ def generate_prompt(image, api_key, style, creativity, neg_prompt, aspect, color
             )
             raw_prompt = response.text
         except Exception as e:
-            return "", None, None, f"🤖 Prompt generation failed: {str(e)}"
-
-        # Step 6: Token Stats
-        input_tokens = len(img_b64) // 4
-        output_tokens = len(raw_prompt.split())
+            return "", f"🤖 Prompt generation failed: {str(e)}"
 
-        return (
-            raw_prompt,
-            {"score": 8, "issues": [], "suggestions": []},
-            {"Input Tokens": input_tokens, "Output Tokens": output_tokens},
-            "✅ Prompt generated successfully!"
-        )
+        return raw_prompt, "✅ Prompt generated successfully!"
 
     except Exception as e:
         traceback.print_exc()
-        return "", None, None, f"💥 Unexpected error: {str(e)}"
-
-
-# --- Modern Copy Function ---
-def copy_text(text):
-    return text, "✓ Copied to clipboard!", gr.Button.update(variant="secondary")
+        return "", f"💥 Unexpected error: {str(e)}"
 
 
 # --- Main Interface ---
 def build_interface():
     global STYLE_INSTRUCTIONS  # Ensure access to global dict
 
-    with gr.Blocks(title="Flux Pro Generator", theme="soft") as app:
+    with gr.Blocks(title="Flux Pro Generator") as app:
         gr.Markdown("# 🎨 Flux Pro Prompt Generator")
         gr.Markdown("Generate optimized design prompts from images using Google's Gemini.")
 
@@ -157,20 +140,12 @@ def build_interface():
 
             with gr.Column(scale=2):
                 status_msg = gr.Textbox(label="📢 Status", interactive=False)
-
                 prompt_output = gr.Textbox(
                     label="📝 Optimized Prompt",
-                    lines=8,
-                    interactive=True,
-                    show_copy_button=True
+                    lines=10,
+                    interactive=True
                 )
 
-                with gr.Row():
-                    copy_btn = gr.Button()#"📋 Copy to Clipboard", variant="secondary")
-
-                quality_report = gr.JSON()#label="🔍 Quality Report")
-                token_stats = gr.JSON()#label="🧮 Token Usage")
-
         # Event Handling
         gen_btn.click(
             fn=generate_prompt,
@@ -178,22 +153,10 @@ def build_interface():
                 img_input, api_key, style, creativity,
                 neg_prompt, aspect, color_mode, dpi
             ],
-            outputs=[prompt_output, quality_report, token_stats, status_msg],
+            outputs=[prompt_output, status_msg],
             api_name="generate"
         )
 
-        copy_btn.click(
-            fn=copy_text,
-            inputs=prompt_output,
-            outputs=[prompt_output, status_msg, copy_btn],
-            js="""
-            (text) => {
-                if(text) navigator.clipboard.writeText(text);
-                return [text, '✓ Copied!', { variant: 'secondary' }];
-            }
-            """
-        )
-
     return app