Sync from GitHub via hub-sync

VERSION

@@ -1 +1 @@
-1e80a581d25c8d350735743e0580c28fdf3fe594
+e53591dafabc331639f010ada7aeec3330d50800

main.py

@@ -39,7 +39,7 @@ AI_BACKEND = _env("AI_BACKEND", "hf").lower()
 AI_MAX_TOKENS = int(_env("AI_MAX_TOKENS", "512"))
 AI_FALLBACK_ORDER = [
     p.strip().lower()
-    for p in _env("AI_FALLBACK_ORDER", "hf,github,openrouter,fireworks").split(",")
+    for p in _env("AI_FALLBACK_ORDER", "hf,google,github,openrouter,fireworks").split(",")
     if p.strip()
 ]
 
@@ -52,6 +52,9 @@ OPENROUTER_MODEL = _env("OPENROUTER_MODEL")
 FIREWORKS_API_KEY = _env("FIREWORKS_API_KEY")
 FIREWORKS_MODEL = _env("FIREWORKS_MODEL")
 
+GOOGLE_API_KEY = _env("GOOGLE_API_KEY")
+GOOGLE_MODEL = _env("GOOGLE_MODEL", "gemini-2.0-flash")
+
 
 # Explicit token passing helps avoid auth ambiguity across local and Space runtimes.
 hf_client = InferenceClient(token=HF_TOKEN) if HF_TOKEN else InferenceClient()
@@ -60,6 +63,7 @@ hf_client = InferenceClient(token=HF_TOKEN) if HF_TOKEN else InferenceClient()
 def _runtime_label() -> str:
     active_model = {
         "hf": HF_MODEL,
+        "google": GOOGLE_MODEL,
         "github": GITHUB_MODEL,
         "openrouter": OPENROUTER_MODEL,
         "fireworks": FIREWORKS_MODEL,
@@ -188,9 +192,61 @@ def _chat_fireworks(messages: list) -> str:
     )
 
 
+def _chat_google(messages: list) -> str:
+    if not GOOGLE_API_KEY:
+        raise ValueError("GOOGLE_API_KEY is missing.")
+    if not GOOGLE_MODEL:
+        raise ValueError("GOOGLE_MODEL is not configured.")
+
+    contents = []
+    for msg in messages:
+        role = msg.get("role")
+        text = str(msg.get("content", ""))
+        if not text:
+            continue
+
+        if role == "assistant":
+            contents.append({"role": "model", "parts": [{"text": text}]})
+        elif role in {"user", "system"}:
+            contents.append({"role": "user", "parts": [{"text": text}]})
+
+    endpoint = (
+        f"https://generativelanguage.googleapis.com/v1beta/models/{GOOGLE_MODEL}:generateContent"
+        f"?key={GOOGLE_API_KEY}"
+    )
+    payload = {
+        "contents": contents,
+        "generationConfig": {"maxOutputTokens": AI_MAX_TOKENS},
+    }
+    request = urllib.request.Request(
+        endpoint,
+        data=json.dumps(payload).encode("utf-8"),
+        headers={"Content-Type": "application/json"},
+        method="POST",
+    )
+
+    try:
+        with urllib.request.urlopen(request, timeout=90) as response:
+            body = json.loads(response.read().decode("utf-8"))
+    except urllib.error.HTTPError as exc:
+        details = exc.read().decode("utf-8", errors="ignore")
+        raise RuntimeError(f"HTTP {exc.code}: {details[:300]}") from exc
+
+    candidates = body.get("candidates") or []
+    if not candidates:
+        raise RuntimeError("No candidates returned from Gemini.")
+
+    parts = (candidates[0].get("content") or {}).get("parts") or []
+    text_chunks = [str(part.get("text", "")) for part in parts if isinstance(part, dict)]
+    result = "".join(text_chunks).strip()
+    return result or "(empty response)"
+
+
 def _chat_once(backend: str, messages: list) -> str:
     if backend == "hf":
         return _chat_hf(messages)
+    if backend == "google":
+        return _chat_google(messages)
     if backend == "github":
         return _chat_github(messages)
     if backend == "openrouter":
@@ -198,7 +254,7 @@ def _chat_once(backend: str, messages: list) -> str:
     if backend == "fireworks":
         return _chat_fireworks(messages)
     raise ValueError(
-        f"Unsupported AI_BACKEND='{backend}'. Use one of: hf, github, openrouter, fireworks, auto"
+        f"Unsupported AI_BACKEND='{backend}'. Use one of: hf, google, github, openrouter, fireworks, auto"
     )
 
 
@@ -228,7 +284,7 @@ with gr.Blocks(title="GitHub + HuggingFace + AI Chat Demo") as demo:
     gr.Markdown("# GitHub → HuggingFace → AI Chat")
     gr.Markdown(f"**{_build_label()}**")
     gr.Markdown(
-        "Multi-provider chat app for learning and testing across HF, GitHub Models, OpenRouter, and Fireworks."
+        "Multi-provider chat app for learning and testing across HF, Gemini, GitHub Models, OpenRouter, and Fireworks."
     )
     gr.Markdown(f"**{_runtime_label()}**")