Upload 3 files

README.md

@@ -1,14 +1,15 @@
 ---
-title: Anycoder 1404a139
-emoji: ⚡
-colorFrom: purple
-colorTo: gray
+title: Anycoder
+emoji: 🤖
+colorFrom: blue
+colorTo: purple
 sdk: gradio
-sdk_version: 6.0.2
 app_file: app.py
 pinned: false
-tags:
-- anycoder
+python_version: "3.10"
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+## Setup
+
+1. Add a Space secret named `GEMINI_API_KEY`.
+2. (Optional) Set `GEMINI_MODEL` to override the default model (`gemini-2.0-flash`).

app.py

@@ -1,175 +1,200 @@
 import os
+import mimetypes
+from typing import Iterable, List, Optional, Tuple, Union
+
 import gradio as gr
-import google.generativeai as genai
-from PIL import Image
-import typing
+from google import genai
+from google.genai import types
+
 
+# -----------------------------
 # Configuration
-MODEL_ID = "gemini-2.0-flash-exp"
+# -----------------------------
+DEFAULT_MODEL = os.getenv("GEMINI_MODEL", "gemini-2.0-flash")
 
-def generate_content(
-    prompt: str,
-    image1: str | None,
-    image2: str | None,
-    image3: str | None,
-    api_key: str
-) -> tuple[str | None, str | None]:
+# Common Gemini models (edit as needed)
+MODEL_CHOICES = [
+    "gemini-2.0-flash",
+    "gemini-2.5-flash",
+    "gemini-2.5-pro",
+    "gemini-3-flash-preview",
+    "gemini-3-pro-preview",
+]
+
+DEFAULT_SYSTEM_INSTRUCTION = (
+    "You are a coding assistant. "
+    "When the user asks for code, output complete, runnable code. "
+    "Prefer simple, readable implementations and include brief comments where helpful."
+)
+
+
+# -----------------------------
+# Client helper
+# -----------------------------
+_client: Optional[genai.Client] = None
+
+
+def _get_api_key() -> Optional[str]:
+    # HF Spaces secrets often use GEMINI_API_KEY; keep a few fallbacks.
+    return (
+        os.getenv("GEMINI_API_KEY")
+        or os.getenv("GOOGLE_API_KEY")
+        or os.getenv("GOOGLE_GENAI_API_KEY")
+    )
+
+
+def get_client() -> genai.Client:
+    global _client
+    if _client is None:
+        api_key = _get_api_key()
+        if not api_key:
+            raise RuntimeError(
+                "Missing API key. Set GEMINI_API_KEY in your Hugging Face Space secrets."
+            )
+        _client = genai.Client(api_key=api_key)
+    return _client
+
+
+# -----------------------------
+# Input helpers
+# -----------------------------
+def _coerce_filepath(item: object) -> Optional[str]:
+    """
+    Gradio file inputs can arrive as:
+      - str paths (type='filepath')
+      - dict-like objects with a 'path' key
+      - objects with a 'path' attribute
+    """
+    if item is None:
+        return None
+    if isinstance(item, str):
+        return item
+    if isinstance(item, dict) and "path" in item:
+        return item["path"]
+    path_attr = getattr(item, "path", None)
+    if isinstance(path_attr, str):
+        return path_attr
+    return None
+
+
+def _guess_mime_type(path: str) -> str:
+    mime, _ = mimetypes.guess_type(path)
+    return mime or "application/octet-stream"
+
+
+def build_contents(prompt: str, file_items: Optional[Iterable[object]]) -> List[object]:
     """
-    Generates content using Google Gemini 2.0 Flash.
-    Handles both text-to-image/image-editing and text responses.
+    Build the `contents` list for client.models.generate_content().
+    Per Google docs, you can pass a list mixing strings and types.Part objects.
     """
-    if not api_key:
-        raise gr.Error("Please enter your Google API Key to continue.")
-    
+    contents: List[object] = [prompt]
+
+    if not file_items:
+        return contents
+
+    for item in file_items:
+        path = _coerce_filepath(item)
+        if not path:
+            continue
+
+        mime_type = _guess_mime_type(path)
+        try:
+            with open(path, "rb") as f:
+                data = f.read()
+        except Exception:
+            # If reading fails, skip that file but continue.
+            continue
+
+        # Attach as binary Part
+        contents.append(types.Part.from_bytes(data=data, mime_type=mime_type))
+
+    return contents
+
+
+# -----------------------------
+# Generation
+# -----------------------------
+def generate(
+    prompt: str,
+    model: str,
+    system_instruction: str,
+    temperature: float,
+    top_p: float,
+    top_k: int,
+    max_output_tokens: int,
+    files: Optional[List[object]],
+) -> str:
+    prompt = (prompt or "").strip()
+    if not prompt:
+        return "Error: prompt is empty."
+
     try:
-        genai.configure(api_key=api_key)
-        model = genai.GenerativeModel(MODEL_ID)
-        
-        # Prepare content parts
-        parts = [prompt]
-        
-        # Add images if they exist
-        images = [image1, image2, image3]
-        for img_path in images:
-            if img_path:
-                # Upload file if necessary or read directly
-                parts.append(Image.open(img_path))
-        
-        # Generate content
-        response = model.generate_content(parts)
-        
-        # Resolve response
-        response.resolve()
-        
-        # Check for image in response
-        image_output = None
-        text_output = None
-        
-        if response.candidates:
-            for part in response.candidates[0].content.parts:
-                if part.inline_data:
-                    # Save the generated image to a temporary file to display it
-                    image_data = part.inline_data.data
-                    import tempfile
-                    temp_dir = tempfile.gettempdir()
-                    # Create a unique filename
-                    filename = os.path.join(temp_dir, f"gemini_output_{os.urandom(4).hex()}.png")
-                    with open(filename, "wb") as f:
-                        f.write(image_data)
-                    image_output = filename
-                elif part.text:
-                    text_output = part.text
-
-        if image_output:
-            return image_output, text_output or "Image generated successfully!"
-        elif text_output:
-            return None, text_output
-        else:
-            return None, "No content generated. Try a different prompt."
+        client = get_client()
+        contents = build_contents(prompt, files)
+
+        config = types.GenerateContentConfig(
+            system_instruction=system_instruction.strip() if system_instruction else None,
+            temperature=temperature,
+            top_p=top_p,
+            top_k=top_k,
+            max_output_tokens=int(max_output_tokens),
+        )
+
+        resp = client.models.generate_content(
+            model=model or DEFAULT_MODEL,
+            contents=contents,
+            config=config,
+        )
 
+        return resp.text or ""
     except Exception as e:
-        raise gr.Error(f"Error generating content: {str(e)}")
+        return f"Error: {type(e).__name__}: {e}"
 
-# Gradio 6 Application
+
+# -----------------------------
+# UI
+# -----------------------------
 with gr.Blocks() as demo:
-    
-    # Header
-    gr.HTML("""
-    <div style="text-align: center; margin-bottom: 20px;">
-        <h1>✨ Gemini 2.0 Flash Studio</h1>
-        <p>Next-Gen Image Generation & Editing powered by Google DeepMind</p>
-        <a href="https://huggingface.co/spaces/akhaliq/anycoder" target="_blank" style="color: #007bff; text-decoration: none; font-weight: bold;">Built with anycoder</a>
-    </div>
-    """)
-    
-    # API Key Section
-    gr.Markdown("### 🔑 Configuration")
+    gr.Markdown("# Anycoder (Gemini via google-genai)\n\n"
+                "This Space uses the **current** Google GenAI Python SDK (`google-genai`).\n"
+                "Set `GEMINI_API_KEY` in Space **Settings → Secrets**.")
+
     with gr.Row():
-        api_key_input = gr.Textbox(
-            label="Google API Key",
-            type="password",
-            placeholder="Enter your 'gemini-2.0-flash-exp' API Key here...",
-            scale=4
+        model = gr.Dropdown(
+            choices=MODEL_CHOICES,
+            value=DEFAULT_MODEL if DEFAULT_MODEL in MODEL_CHOICES else MODEL_CHOICES[0],
+            label="Model",
         )
-    
-    gr.Markdown("### 🎨 Inputs")
-    
-    # Image Inputs (3 Blank Spots)
+        temperature = gr.Slider(0.0, 2.0, value=0.2, step=0.05, label="Temperature")
+        max_output_tokens = gr.Slider(128, 8192, value=2048, step=64, label="Max output tokens")
+
     with gr.Row():
-        img_input_1 = gr.Image(
-            label="Image 1 (Drag & Drop)", 
-            type="filepath", 
-            sources=["upload", "clipboard"],
-            height=300
-        )
-        img_input_2 = gr.Image(
-            label="Image 2 (Drag & Drop)", 
-            type="filepath", 
-            sources=["upload", "clipboard"],
-            height=300
-        )
-        img_input_3 = gr.Image(
-            label="Image 3 (Drag & Drop)", 
-            type="filepath", 
-            sources=["upload", "clipboard"],
-            height=300
-        )
+        top_p = gr.Slider(0.0, 1.0, value=0.95, step=0.01, label="Top-p")
+        top_k = gr.Slider(1, 100, value=40, step=1, label="Top-k")
 
-    # Prompt Input
-    prompt_input = gr.Textbox(
-        label="✍️ Prompt",
-        placeholder="Describe the image you want to generate or how you want to edit the uploaded images...",
+    system_instruction = gr.Textbox(
+        label="System instruction (optional)",
+        value=DEFAULT_SYSTEM_INSTRUCTION,
         lines=3,
-        show_copy_button=True
     )
-    
-    # Action Button
-    generate_btn = gr.Button("✨ Generate", variant="primary", size="lg")
-    
-    gr.Markdown("### 🖼️ Output")
-    
-    # Output Section
-    with gr.Row():
-        with gr.Column():
-            output_image = gr.Image(label="Generated Image", type="filepath", height=400)
-        with gr.Column():
-            output_text = gr.Textbox(label="Response / Description", lines=10)
-
-    # Examples
-    gr.Examples(
-        examples=[
-            ["Generate a futuristic cyberpunk city with neon lights", None, None, None],
-            ["Edit this image to make it look like a painting", "https://gradio-builds.s3.amazonaws.com/assets/cheetah-003.jpg", None, None],
-            ["Combine these two concepts into a logo", None, None, None], # Placeholder for logic
-        ],
-        inputs=[prompt_input, img_input_1, img_input_2, img_input_3],
+
+    prompt = gr.Textbox(label="Prompt", lines=8, placeholder="Describe what you want the model to generate...")
+
+    files = gr.Files(
+        label="Optional files (images, audio, etc.)",
+        file_count="multiple",
+        type="filepath",
     )
 
-    # Event Listener
-    generate_btn.click(
-        fn=generate_content,
-        inputs=[prompt_input, img_input_1, img_input_2, img_input_3, api_key_input],
-        outputs=[output_image, output_text],
-        api_visibility="public"
+    btn = gr.Button("Generate", variant="primary")
+    output = gr.Code(label="Model output", language="markdown")
+
+    btn.click(
+        fn=generate,
+        inputs=[prompt, model, system_instruction, temperature, top_p, top_k, max_output_tokens, files],
+        outputs=[output],
+        api_visibility="public",
     )
 
-# Gradio 6 Launch Configuration
-# Note: theme, css, and footer_links are now in launch()
-demo.launch(
-    theme=gr.themes.Soft(
-        primary_hue="indigo",
-        secondary_hue="cyan",
-        neutral_hue="slate",
-        font=gr.themes.GoogleFont("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_text_color="white",
-        block_border_width="0px",
-        block_shadow="*shadow_drop_lg"
-    ),
-    footer_links=[{"label": "Built with anycoder", "url": "https://huggingface.co/spaces/akhaliq/anycoder"}]
-)
+if __name__ == "__main__":
+    demo.queue(api_open=False).launch(show_api=False)

requirements.txt

@@ -1,3 +1,2 @@
-Pillow
-google
 gradio>=6.0.2
+google-genai