Real gateway: chat completions + Drive storage

Dockerfile

@@ -0,0 +1,14 @@
+FROM python:3.10
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,73 @@
+from fastapi import FastAPI, UploadFile, File
+from fastapi.middleware.cors import CORSMiddleware
+import httpx, io, json, os
+from googleapiclient.discovery import build
+from googleapiclient.http import MediaIoBaseUpload
+from google.oauth2 import service_account
+
+app = FastAPI()
+app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_methods=["*"], allow_headers=["*"])
+
+DRIVE_FOLDER_ID = os.environ.get("DRIVE_FOLDER_ID", "")
+_creds = None
+drive = None
+
+def get_drive():
+    global _creds, drive
+    if drive is None:
+        _creds = service_account.Credentials.from_service_account_info(
+            json.loads(os.environ["GOOGLE_SERVICE_ACCOUNT_JSON"]),
+            scopes=["https://www.googleapis.com/auth/drive"]
+        )
+        drive = build("drive", "v3", credentials=_creds)
+    return drive
+
+@app.get("/")
+def root():
+    return {"status": "Dolor D Prince API Gateway running"}
+
+@app.post("/v1/chat/completions")
+async def chat_completions(body: dict):
+    api_key = os.environ.get("OPENROUTER_API_KEY", "")
+    async with httpx.AsyncClient(timeout=60) as client:
+        res = await client.post(
+            "https://openrouter.ai/api/v1/chat/completions",
+            headers={"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"},
+            json=body,
+        )
+        return res.json()
+
+@app.post("/v1/projects/{page_id}")
+async def save_project(page_id: str, body: dict):
+    d = get_drive()
+    fname = f"{page_id}.json"
+    media = MediaIoBaseUpload(io.BytesIO(json.dumps(body).encode()), mimetype="application/json")
+    existing = d.files().list(q=f"name='{fname}' and '{DRIVE_FOLDER_ID}' in parents").execute()
+    if existing["files"]:
+        d.files().update(fileId=existing["files"][0]["id"], media_body=media).execute()
+    else:
+        d.files().create(body={"name": fname, "parents": [DRIVE_FOLDER_ID]}, media_body=media).execute()
+    return {"status": "ok"}
+
+@app.get("/v1/projects/{page_id}")
+async def load_project(page_id: str):
+    d = get_drive()
+    fname = f"{page_id}.json"
+    res = d.files().list(q=f"name='{fname}' and '{DRIVE_FOLDER_ID}' in parents").execute()
+    if not res["files"]:
+        return {"data": {}}
+    content = d.files().get_media(fileId=res["files"][0]["id"]).execute()
+    return {"data": json.loads(content)}
+
+@app.post("/v1/assets/upload")
+async def upload_asset(files: list[UploadFile] = File(...)):
+    d = get_drive()
+    urls = []
+    for f in files:
+        media = MediaIoBaseUpload(io.BytesIO(await f.read()), mimetype=f.content_type)
+        created = d.files().create(
+            body={"name": f.filename, "parents": [DRIVE_FOLDER_ID]}, media_body=media, fields="id"
+        ).execute()
+        d.permissions().create(fileId=created["id"], body={"role": "reader", "type": "anyone"}).execute()
+        urls.append(f"https://drive.google.com/uc?id={created['id']}")
+    return {"data": [{"src": u} for u in urls]}

requirements.txt

@@ -0,0 +1,7 @@
+fastapi
+uvicorn[standard]
+httpx
+python-multipart
+google-api-python-client
+google-auth
+pydantic