Update app.py

app.py

@@ -2,88 +2,101 @@ import os
 import gradio as gr
 import google.generativeai as genai
 from PIL import Image
+import io
+import time
 
-# Configuration for Hugging Face Spaces
-MAX_SIZE = 512  # Reduced image size for HF memory limits
-TIMEOUT = 18  # Stay safely under 20s timeout
-STYLES = ["General", "Vector", "Realistic", "Kawaii"]
+# Hugging Face-safe configuration
+MAX_SIZE = 768
+TIMEOUT = 15
+CONCURRENCY = 2  # Reduced for HF stability
 
-def process_image(image):
-    """Optimize image for Hugging Face environment"""
-    img = image.convert("RGB")
-    img.thumbnail((MAX_SIZE, MAX_SIZE))
-    return img
+STYLE_INSTRUCTIONS = {
+    "General": "Describe this design concisely for Flux",
+    "Vector": "Clean vector style with sharp lines",
+    "Realistic": "Photorealistic details",
+    "Kawaii": "Cute cartoon style with pastel colors"
+}
 
-def generate_prompt(image, api_key, style):
+def hf_safe_generate(image, api_key, style):
     try:
-        # Validate inputs
-        if not image:
-            return "⚠️ Please upload an image first"
-        if not api_key:
-            return "🔑 API key required"
-
-        # Process image
-        img = process_image(image)
+        start_time = time.time()
+        
+        # Convert and resize image
+        img = image.convert("RGB")
+        img.thumbnail((MAX_SIZE, MAX_SIZE))
+        buffer = io.BytesIO()
+        img.save(buffer, format="JPEG", quality=85)
         
-        # Configure Gemini
+        # Configure API
         genai.configure(api_key=api_key)
         model = genai.GenerativeModel('gemini-1.5-pro')
         
-        # Create prompt based on style
-        base_prompt = "Describe this design concisely for Flux 1.1 Pro"
-        if style == "Vector":
-            instruction = f"{base_prompt} in clean vector style with sharp lines"
-        elif style == "Realistic":
-            instruction = f"{base_prompt} with photorealistic details"
-        else:
-            instruction = base_prompt
-
-        # Generate with timeout safety
+        # Timeout check
+        if time.time() - start_time > 5:
+            return "Error: Image processing too slow"
+            
         response = model.generate_content(
-            contents=[instruction, img],
+            [STYLE_INSTRUCTIONS[style], 
+            {"mime_type": "image/jpeg", "data": buffer.getvalue()},
             request_options={"timeout": TIMEOUT}
         )
         
-        return response.text if response.text else "❌ No response generated"
-
+        return response.text[:500] if response.text else "No response"
+        
     except Exception as e:
-        return f"⚠️ Error: {str(e)}"
+        return f"Error: {str(e)}"
 
-# Gradio interface optimized for Hugging Face
-with gr.Blocks(title="Flux Prompt Generator") as app:
-    gr.Markdown("# 🎨 Flux AI Prompt Generator")
+# Enhanced interface with original features
+with gr.Blocks(title="Flux Pro HF", theme=gr.themes.Soft()) as app:
+    gr.Markdown("## 🚀 **Flux Prompt Generator**")
     
     with gr.Row():
-        api_key = gr.Textbox(
-            label="Google Gemini API Key",
-            type="password",
-            placeholder="Enter your API key..."
-        )
-        style = gr.Dropdown(
-            STYLES,
-            value="General",
-            label="Design Style"
-        )
-    
-    image_input = gr.Image(
-        label="Upload Your Design",
-        type="pil",
-        height=300
-    )
-    
-    generate_btn = gr.Button("Generate Prompt", variant="primary")
-    
+        with gr.Column():
+            api_key = gr.Textbox(
+                label="🔑 Gemini API Key",
+                type="password",
+                placeholder="Enter your API key..."
+            )
+            style = gr.Dropdown(
+                list(STYLE_INSTRUCTIONS.keys()),
+                value="General",
+                label="🎨 Design Style",
+                interactive=True
+            )
+            image = gr.Image(
+                label="🖼️ Upload Design",
+                type="pil",
+                height=300
+            )
+            
+    with gr.Row():
+        generate_btn = gr.Button("✨ Generate Prompt", variant="primary")
+        copy_btn = gr.Button("📋 Copy")
+        
     output = gr.Textbox(
-        label="Generated Prompt",
-        placeholder="Your prompt will appear here...",
-        lines=5
+        label="📝 Generated Prompt",
+        lines=5,
+        max_lines=8,
+        interactive=False
     )
-
+    
+    # Event handlers
     generate_btn.click(
-        fn=generate_prompt,
-        inputs=[image_input, api_key, style],
+        hf_safe_generate,
+        inputs=[image, api_key, style],
         outputs=output
     )
+    
+    copy_btn.click(
+        lambda x: x,
+        inputs=output,
+        outputs=output,
+        _js="(text) => navigator.clipboard.writeText(text)"
+    )
 
-# Hugging Face compatible launch settings
-app.launch(debug=False, show_error=True)
+# Fixed queue configuration for Hugging Face
+app.queue(concurrency_count=CONCURRENCY).launch(
+    debug=False,
+    max_threads=1,
+    show_error=True
+)