Update app.py from anycoder

app.py

@@ -1,87 +1,127 @@
 import gradio as gr
 import time
 import numpy as np
-from PIL import Image, ImageEnhance
+from PIL import Image, ImageEnhance, ImageFilter
 import random
 
-def simulate_gemini_flash_processing(image_editor_data, prompt, mode, strength):
+def gemini_chatbot_processor(image_input, chat_history, prompt, mode, strength):
     """
-    Simulates the behavior of a high-speed AI image editing model (like Gemini 2.0 Flash).
-    Since we cannot connect to the actual paid/private API without a key, 
-    this function simulates the processing latency and applies a visual transformation
-    to demonstrate the UI flow.
+    Simulates a GPT-ChatBot experience powered by 'Gemini 2.0 Flash'.
+    Handles both Image Generation (Text-to-Image) and Image Editing (Image-to-Image).
     """
-    if image_editor_data is None:
-        gr.Warning("Please upload an image first!")
-        return None
     
-    # Simulate "Flash" speed inference
-    yield gr.Info("🚀 Gemini 2.0 Flash is processing...")
-    time.sleep(1.2)
+    # 1. Update Chat History with User Message
+    if not prompt:
+        gr.Warning("Please enter a prompt.")
+        return chat_history, image_input
+
+    chat_history = chat_history + [[prompt, None]]
     
-    # Extract the composite image (background + layers + brush strokes)
-    # In a real app, this would be sent to the API with the prompt
+    # 2. Simulate AI "Thinking" (Streaming effect simulation)
+    time.sleep(0.5)
+    chat_history[-1][1] = "🤔 Thinking..."
+    yield chat_history, None
     
-    if isinstance(image_editor_data, dict) and "composite" in image_editor_data:
-        img = image_editor_data["composite"]
-    elif isinstance(image_editor_data, np.ndarray):
-        img = image_editor_data
-    else:
-        # Fallback for different data structures
-        return image_editor_data
-
-    # Convert to PIL for processing
-    if isinstance(img, np.ndarray):
-        pil_img = Image.fromarray(img)
-    else:
-        pil_img = img
-
-    # --- SIMULATED AI EFFECTS ---
-    # We apply deterministic but varied effects based on the prompt hash or mode
-    # to show that "something happened".
+    time.sleep(0.8)
+    
+    # 3. Determine Logic: Generation vs Editing
+    result_image = None
+    response_text = ""
     
-    width, height = pil_img.size
+    # Check if user uploaded an image or wants to generate new
+    has_image = image_input is not None
     
-    # 1. Simple "Refinement" (Sharpening/Contrast)
-    if mode == "Refinement":
-        enhancer = ImageEnhance.Contrast(pil_img)
-        pil_img = enhancer.enhance(1.1 + (strength * 0.2))
-        enhancer = ImageEnhance.Sharpness(pil_img)
-        pil_img = enhancer.enhance(1.1)
+    if mode == "Generate New Image" or not has_image:
+        # --- TEXT-TO-IMAGE GENERATION SIMULATION ---
+        chat_history[-1][1] = "✨ Generating new image based on your prompt..."
+        
+        # Simulate processing time
+        time.sleep(1.0)
+        
+        # Create a procedural "generated" image based on prompt hash (deterministic random)
+        # This simulates the AI creating something unique based on text
+        width, height = 512, 512
+        
+        # Create a gradient/pattern based on prompt string
+        hash_val = hash(prompt)
+        random.seed(hash_val)
+        
+        # Create a base color
+        r = random.randint(50, 200)
+        g = random.randint(50, 200)
+        b = random.randint(50, 200)
+        
+        img = Image.new("RGB", (width, height), (r, g, b))
+        pixels = img.load()
+        
+        # Add some "AI" noise/pattern
+        for i in range(width):
+            for j in range(height):
+                noise = random.randint(-20, 20)
+                pixels[i, j] = (
+                    max(0, min(255, r + noise)),
+                    max(0, min(255, g + noise)),
+                    max(0, min(255, b + noise))
+                )
+        
+        # Add a geometric pattern to look like a "generated" abstract art
+        img = img.filter(ImageFilter.SMOOTH)
+        
+        result_image = np.array(img)
+        response_text = f"Here is a new image generated for: '{prompt}'"
         
-    # 2. Simple "Style Transfer" (Simulated via color shift/filter)
-    elif mode == "Style Transfer":
-        # Split channels and shift them slightly
-        r, g, b = pil_img.split()
-        r = r.point(lambda i: i * 1.05)
-        b = b.point(lambda i: i * 0.95)
-        pil_img = Image.merge("RGB", (r, g, b))
+    else:
+        # --- IMAGE EDITING SIMULATION ---
+        chat_history[-1][1] = "🎨 Analyzing image and applying edits..."
+        time.sleep(1.2)
         
-    # 3. "Inpainting" / "Magic Fill" (Simulated via slight blur/smooth in masked areas logic)
-    # Since we don't have the mask explicitly separated in the composite easily without complex logic,
-    # we'll just apply a brightness boost to signify "filling light".
-    elif mode == "Inpainting":
-        enhancer = ImageEnhance.Brightness(pil_img)
-        pil_img = enhancer.enhance(1.05 + (strength * 0.1))
+        # Extract image
+        if isinstance(image_input, dict) and "composite" in image_input:
+            pil_img = Image.fromarray(image_input["composite"])
+        elif isinstance(image_input, np.ndarray):
+            pil_img = Image.fromarray(image_input)
+        else:
+            pil_img = image_input
+
+        # Apply simulated effects
+        if mode == "Enhance / Refine":
+            factor = 1.0 + (strength * 0.5)
+            pil_img = ImageEnhance.Contrast(pil_img).enhance(factor)
+            pil_img = ImageEnhance.Sharpness(pil_img).enhance(1.2)
+            response_text = "I've enhanced the sharpness and contrast of your image."
+            
+        elif mode == "Change Style":
+            # Simulate style transfer with a tint
+            r, g, b = pil_img.split()
+            r = r.point(lambda i: i * 1.1)
+            pil_img = Image.merge("RGB", (r, g, b))
+            response_text = "I've applied a stylistic filter to the image."
+            
+        elif mode == "Magic Fill / Inpaint":
+            # Simulate inpainting by slightly brightening masked areas (simulated logic)
+            pil_img = ImageEnhance.Brightness(pil_img).enhance(1.1)
+            response_text = "I've filled in the areas and adjusted the lighting."
 
-    # Convert back to numpy for Gradio
-    result = np.array(pil_img)
+        result_image = np.array(pil_img)
+
+    # 4. Final Chat Update
+    chat_history[-1][1] = response_text
     
-    gr.Success("✨ Generation Complete!")
-    return result
+    gr.Success("✨ Operation Complete!")
+    return chat_history, result_image
 
 # --- Gradio 6 Application Setup ---
 
 with gr.Blocks() as demo:
     
-    # Header Section
+    # Header with Branding
     gr.HTML("""
-        <div style="text-align: center; margin-bottom: 20px; border-bottom: 1px solid #e5e7eb; padding-bottom: 20px;">
-            <h1 style="margin: 0; font-size: 2.5rem; background: linear-gradient(90deg, #FF6B6B, #FF8E53); -webkit-background-clip: text; -webkit-text-fill-color: transparent;">
-                Gemini 2.0 Flash Live Image Editor
+        <div style="text-align: center; margin-bottom: 25px; border-bottom: 1px solid #e5e7eb; padding-bottom: 20px;">
+            <h1 style="margin: 0; font-size: 2.5rem; background: linear-gradient(90deg, #4285F4, #34A853, #FBBC05, #EA4335); -webkit-background-clip: text; -webkit-text-fill-color: transparent;">
+                GPT-ChatBot x Gemini 2.0 Flash
             </h1>
-            <p style="color: #6b7280; margin-top: 10px;">
-                Unlimited Online Free Demo • Upload, Sketch, and Edit with AI
+            <p style="color: #6b7280; margin-top: 10px; font-size: 1.1rem;">
+                Unlimited Online Free Demo • AI Image Generation & Editing
             </p>
             <div style="margin-top: 15px;">
                 <a href="https://huggingface.co/spaces/akhaliq/anycoder" target="_blank" 
@@ -95,113 +135,117 @@ with gr.Blocks() as demo:
     """)
 
     with gr.Row():
-        # Left Column: Canvas
+        # Left Column: Editor & Canvas
         with gr.Column(scale=3):
+            gr.Markdown("### 🖼️ Canvas")
             editor = gr.ImageEditor(
-                label="Interactive Canvas",
+                label="Upload or Sketch",
                 type="numpy",
                 layers=True,
                 sources=["upload", "webcam", "clipboard"],
-                transforms=["crop"],
+                transforms=["crop", "rotate"],
                 brush=gr.Brush(
-                    colors=["#FFFFFF", "#000000", "#FF0000", "#00FF00", "#0000FF", "#FFFF00"],
-                    default_size="20",
+                    colors=["#FFFFFF", "#000000", "#FF0000", "#00FF00"],
+                    default_size="15",
                     color_mode="fixed"
                 ),
                 eraser=gr.Eraser(default_size="20"),
-                height=600
+                height=550
             )
             
-            # Instructions
-            gr.Markdown("""
-            **How to use:**
-            1. Upload an image or take a photo.
-            2. Use the **Brush** to mark areas you want to change.
-            3. Type your instruction in the box on the right.
-            4. Click **Generate**.
-            """)
-
-        # Right Column: Controls
-        with gr.Column(scale=1):
-            gr.Markdown("### ⚙️ AI Controls")
-            
-            prompt_input = gr.Textbox(
-                label="Instruction Prompt",
-                placeholder="e.g. 'Remove the background', 'Make it look like a painting', 'Fix the lighting'",
-                lines=3,
-                interactive=True
+            # Output Image Display
+            output_image = gr.Image(
+                label="Result",
+                interactive=False,
+                show_download_button=True,
+                height=400
             )
+
+        # Right Column: Chatbot & Controls
+        with gr.Column(scale=2):
+            gr.Markdown("### 🤖 AI Chat Interface")
             
-            mode = gr.Radio(
-                choices=["Inpainting", "Refinement", "Style Transfer"],
-                value="Inpainting",
-                label="Processing Mode",
-                info="Select the type of AI modification"
+            # Chatbot History
+            chatbot = gr.Chatbot(
+                label="Conversation History",
+                height=300,
+                show_copy_button=True,
+                avatar_images=(None, "https://www.gstatic.com/lamda/images/gemini_sparkle_v002_d47a511478e8.svg") # Generic AI avatar
             )
             
-            strength = gr.Slider(
-                minimum=0.1,
-                maximum=1.0,
-                value=0.7,
-                step=0.1,
-                label="Effect Strength"
-            )
+            # Controls
+            with gr.Accordion("⚙️ Generation Settings", open=True):
+                prompt_input = gr.Textbox(
+                    label="Your Instruction",
+                    placeholder="Describe the image you want to generate or how to edit the current one...",
+                    lines=3,
+                    autofocus=True
+                )
+                
+                mode = gr.Radio(
+                    choices=["Generate New Image", "Enhance / Refine", "Change Style", "Magic Fill / Inpaint"],
+                    value="Enhance / Refine",
+                    label="Operation Mode",
+                    info="Choose to create new or edit existing"
+                )
+                
+                strength = gr.Slider(
+                    minimum=0.1,
+                    maximum=1.0,
+                    value=0.7,
+                    step=0.1,
+                    label="Effect Intensity"
+                )
 
             generate_btn = gr.Button(
-                "✨ Generate with Flash AI",
+                "🚀 Send to Gemini Flash",
                 variant="primary",
                 size="lg"
             )
             
-            # Output Preview
-            result_image = gr.Image(
-                label="AI Result",
-                interactive=False,
-                show_download_button=True
-            )
-            
-            with gr.Accordion("Advanced Settings", open=False):
-                seed = gr.Number(label="Seed (Random)", value=-1, precision=0)
-                steps = gr.Slider(minimum=1, maximum=50, value=25, step=1, label="Inference Steps")
+            clear_btn = gr.ClearButton([prompt_input, chatbot, output_image], value="🗑️ Clear History")
 
-    # Examples Section
+    # Examples
     gr.Examples(
         examples=[
-            ["https://images.unsplash.com/photo-1542204165-65bf26472b9b?w=600", "Make the sky more dramatic", "Refinement", 0.8],
-            ["https://images.unsplash.com/photo-1506744038136-46273834b3fb?w=600", "Apply a vintage filter", "Style Transfer", 0.6],
+            [None, "A futuristic cyberpunk city at night with neon lights", "Generate New Image", 0.8],
+            [None, "A cute oil painting of a cat in a garden", "Generate New Image", 0.7],
+            ["https://images.unsplash.com/photo-1506744038136-46273834b3fb?w=600", "Make this look like a vintage photo", "Change Style", 0.9],
         ],
         inputs=[editor, prompt_input, mode, strength],
         cache_examples=False,
-        label="Try these examples"
+        label="Quick Start Examples"
     )
 
     # Event Listeners
     generate_btn.click(
-        fn=simulate_gemini_flash_processing,
-        inputs=[editor, prompt_input, mode, strength],
-        outputs=[result_image],
+        fn=gemini_chatbot_processor,
+        inputs=[editor, chatbot, prompt_input, mode, strength],
+        outputs=[chatbot, output_image],
         api_visibility="public"
     )
 
 # Launch Configuration (Gradio 6 Syntax)
 demo.launch(
     theme=gr.themes.Soft(
-        primary_hue="orange",  # "Flash" theme color
-        secondary_hue="blue",
-        neutral_hue="gray",
-        font=gr.themes.GoogleFont("Inter"),
+        primary_hue="blue",   # Google Blue
+        secondary_hue="green", # Google Green
+        neutral_hue="slate",
+        font=gr.themes.GoogleFont("Google Sans", "Inter"),
         text_size="lg",
         spacing_size="lg",
         radius_size="md"
     ).set(
-        button_primary_background_fill="*primary_500",
-        button_primary_background_fill_hover="*primary_600",
+        button_primary_background_fill="*primary_600",
+        button_primary_background_fill_hover="*primary_700",
         button_primary_text_color="white",
-        block_background_fill="*background_fill_secondary",
+        # Custom Chatbot styling
+        chatbot_message_background_fill_user="*primary_100",
+        chatbot_message_background_fill_bot="*background_fill_secondary",
     ),
     footer_links=[
         {"label": "Built with anycoder", "url": "https://huggingface.co/spaces/akhaliq/anycoder"},
         {"label": "Gradio", "url": "https://gradio.app"},
-        {"label": "View API", "url": "https://gradio.app"} # Placeholder for internal API link
+        {"label": "Google DeepMind", "url": "https://deepmind.google"}
     ]
 )