Update app.py

app.py

@@ -1,19 +1,226 @@
-from fastapi import FastAPI
-from fastapi.responses import HTMLResponse
+import os
+import json
+import requests
+from fastapi import FastAPI, Request, Header
+from fastapi.responses import HTMLResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
-from fastapi import Request
-import os
 
-app = FastAPI(title="PyRunner AI Space")
+# =========================
+# APP INIT
+# =========================
+
+app = FastAPI(title="PyRunner AI Space", version="1.0")
 
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 
 templates = Jinja2Templates(directory=os.path.join(BASE_DIR, "templates"))
 
-app.mount("/static", StaticFiles(directory=os.path.join(BASE_DIR, "static")), name="static")
+static_path = os.path.join(BASE_DIR, "static")
+if os.path.exists(static_path):
+    app.mount("/static", StaticFiles(directory=static_path), name="static")
+
+
+# =========================
+# ENV
+# =========================
+
+OLLAMA_URL = os.getenv("OLLAMA_URL", "http://localhost:11434")
+OLLAMA_FALLBACK_URL = os.getenv("OLLAMA_FALLBACK_URL")
+
+OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "qwen2.5:1.5b")
+OLLAMA_CODE_MODEL = os.getenv("OLLAMA_CODE_MODEL", "qwen2.5-coder:1.5b")
+OLLAMA_FAST_MODEL = os.getenv("OLLAMA_FAST_MODEL", "deepseek-coder:1.3b")
+OLLAMA_BACKUP_MODEL = os.getenv("OLLAMA_BACKUP_MODEL", "llama3.2:1b")
+
+OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
+OPENROUTER_MODEL = os.getenv("OPENROUTER_MODEL")
+
+SUPABASE_URL = os.getenv("SUPABASE_URL")
+SUPABASE_KEY = os.getenv("SUPABASE_KEY")
+
+
+# =========================
+# SAFE OLLAMA CALL
+# =========================
+
+def call_ollama(prompt: str, model: str):
+    try:
+        r = requests.post(
+            f"{OLLAMA_URL}/api/generate",
+            json={
+                "model": model,
+                "prompt": prompt,
+                "stream": False
+            },
+            timeout=60
+        )
+        return r.json()
+    except Exception as e:
+        return {"error": str(e)}
+
 
+# =========================
+# OPENROUTER FALLBACK
+# =========================
+
+def call_openrouter(prompt: str):
+    if not OPENROUTER_API_KEY:
+        return {"error": "No OpenRouter key"}
+
+    try:
+        r = requests.post(
+            "https://openrouter.ai/api/v1/chat/completions",
+            headers={
+                "Authorization": f"Bearer {OPENROUTER_API_KEY}",
+                "Content-Type": "application/json"
+            },
+            json={
+                "model": OPENROUTER_MODEL,
+                "messages": [
+                    {"role": "user", "content": prompt}
+                ]
+            },
+            timeout=60
+        )
+        return r.json()
+    except Exception as e:
+        return {"error": str(e)}
+
+
+# =========================
+# ROUTER (1B–2B ONLY LOGIC)
+# =========================
+
+def route_model(task: str):
+    task = task.lower()
+
+    if "code" in task or "script" in task:
+        return OLLAMA_CODE_MODEL
+
+    if "fast" in task:
+        return OLLAMA_FAST_MODEL
+
+    if "fallback" in task:
+        return OLLAMA_BACKUP_MODEL
+
+    return OLLAMA_MODEL
+
+
+# =========================
+# UI ROUTE
+# =========================
 
 @app.get("/", response_class=HTMLResponse)
 async def home(request: Request):
-    return templates.TemplateResponse("index.html", {"request": request})
+    return templates.TemplateResponse("index.html", {"request": request})
+
+
+# =========================
+# HEALTH CHECK
+# =========================
+
+@app.get("/health")
+def health():
+    return {
+        "status": "ok",
+        "ollama": OLLAMA_URL,
+        "models": {
+            "default": OLLAMA_MODEL,
+            "code": OLLAMA_CODE_MODEL
+        }
+    }
+
+
+# =========================
+# CHAT ENDPOINT
+# =========================
+
+@app.post("/api/chat")
+async def chat(request: Request):
+    body = await request.json()
+
+    prompt = body.get("prompt", "")
+    mode = body.get("mode", "default")
+
+    model = route_model(mode)
+
+    result = call_ollama(prompt, model)
+
+    # fallback logic
+    if "error" in result and OLLAMA_FALLBACK_URL:
+        try:
+            r = requests.post(
+                f"{OLLAMA_FALLBACK_URL}/api/generate",
+                json={
+                    "model": OLLAMA_BACKUP_MODEL,
+                    "prompt": prompt,
+                    "stream": False
+                },
+                timeout=60
+            )
+            result = r.json()
+        except Exception as e:
+            result = {"error": str(e)}
+
+    return JSONResponse(result)
+
+
+# =========================
+# IMAGE PIPELINE HOOK (stub)
+# =========================
+
+@app.post("/api/image")
+async def image_pipeline(request: Request):
+    body = await request.json()
+
+    prompt = body.get("prompt", "a modern hairstyle")
+
+    return {
+        "status": "queued",
+        "message": "Image pipeline not yet connected",
+        "prompt": prompt
+    }
+
+
+# =========================
+# N8N WEBHOOK TRIGGER
+# =========================
+
+@app.post("/api/webhook/n8n")
+async def trigger_n8n(request: Request):
+    body = await request.json()
+
+    webhook_url = os.getenv("N8N_WEBHOOK_URL")
+
+    if not webhook_url:
+        return {"error": "No webhook configured"}
+
+    try:
+        r = requests.post(webhook_url, json=body, timeout=30)
+        return {"status": "sent", "response": r.text}
+    except Exception as e:
+        return {"error": str(e)}
+
+
+# =========================
+# DEBUG CONFIG (SAFE)
+# =========================
+
+@app.get("/config")
+def config():
+    return {
+        "ollama_url": OLLAMA_URL,
+        "models": {
+            "default": OLLAMA_MODEL,
+            "code": OLLAMA_CODE_MODEL,
+            "fast": OLLAMA_FAST_MODEL,
+            "backup": OLLAMA_BACKUP_MODEL
+        },
+        "openrouter": {
+            "enabled": bool(OPENROUTER_API_KEY)
+        },
+        "supabase": {
+            "enabled": bool(SUPABASE_URL and SUPABASE_KEY)
+        }
+    }