Spaces:

Kleinpuki2
/

ai1

Sleeping

App Files Files Community

Kleinpuki2 commited on May 4

Commit

f2f562e

verified ·

1 Parent(s): 603f4e3

Upload 3 files

Browse files

Files changed (3) hide show

app.py +82 -0
requirements.txt +7 -0
templates/index.html +86 -0

app.py ADDED Viewed

	@@ -0,0 +1,82 @@

+import os
+import torch
+import json
+import uuid
+from fastapi import FastAPI, Request, HTTPException, Depends
+from fastapi.responses import HTMLResponse, JSONResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
+from model import MiniTransformer, BPETokenizer
+app = FastAPI()
+templates = Jinja2Templates(directory="templates")
+# --- MODEL SETUP ---
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+MODEL_PATH = "madgames_gpt2_stable.pth" # Wird beim Deployment geladen
+tokenizer = BPETokenizer()
+# Lädt das Modell (Dummy-Check falls Datei noch nicht da)
+model = None
+if os.path.exists(MODEL_PATH):
+    ckpt = torch.load(MODEL_PATH, map_location=DEVICE, weights_only=False)
+    config = ckpt["config"]
+    model = MiniTransformer(
+        vocab_size=config["vocab_size"],
+        emb_dim=config["emb_dim"],
+        n_layers=config["n_layers"],
+        n_heads=config["n_heads"],
+        ctx_len=config["ctx_len"]
+    ).to(DEVICE)
+    model.load_state_dict(ckpt["model_state"])
+    model.eval()
+# --- KEY MANAGEMENT ---
+KEYS_FILE = "api_keys.json"
+def load_keys():
+    if not os.path.exists(KEYS_FILE): return {}
+    with open(KEYS_FILE, "r") as f: return json.load(f)
+def save_keys(keys):
+    with open(KEYS_FILE, "w") as f: json.dump(keys, f)
+# --- ROUTES ---
+@app.get("/", response_class=HTMLResponse)
+async def dashboard(request: Request):
+    return templates.TemplateResponse("index.html", {"request": request})
+@app.post("/generate_key")
+async def create_key():
+    new_key = f"MG-{uuid.uuid4().hex[:12].upper()}"
+    keys = load_keys()
+    keys[new_key] = {"created": str(torch.utils.data.dataloader.default_collate), "uses": 0}
+    save_keys(keys)
+    return {"key": new_key}
+@app.post("/predict")
+async def predict(request: Request):
+    data = await request.json()
+    api_key = request.headers.get("X-API-Key")
+    keys = load_keys()
+    if api_key not in keys:
+        raise HTTPException(status_code=403, detail="Ungültiger API-Key")
+    prompt = data.get("prompt", "")
+    if not model or not prompt:
+        return {"response": "Server bereit, aber Modell oder Prompt fehlt."}
+    # Generierung (stark vereinfacht für Demo)
+    tokens = torch.tensor(tokenizer.encode(prompt)).unsqueeze(0).to(DEVICE)
+    with torch.no_grad():
+        out = model.generate(tokens, max_new_tokens=100)[0].tolist()
+    keys[api_key]["uses"] += 1
+    save_keys(keys)
+    return {"response": tokenizer.decode(out)}
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)

requirements.txt ADDED Viewed

	@@ -0,0 +1,7 @@

+fastapi
+uvicorn
+torch
+numpy
+python-multipart
+jinja2
+aiofiles

templates/index.html ADDED Viewed

	@@ -0,0 +1,86 @@

+<!DOCTYPE html>
+<html lang="de">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>MadGames AI | API Dashboard</title>
+    <style>
+        :root {
+            --accent: #ff6b00;
+            --bg: #0a0a0c;
+            --glass: rgba(255, 255, 255, 0.03);
+            --border: rgba(255, 255, 255, 0.08);
+        }
+        body {
+            font-family: 'Inter', system-ui, sans-serif;
+            background: var(--bg);
+            color: #fff;
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            height: 100vh;
+            margin: 0;
+            background-image: radial-gradient(circle at 50% 50%, #1a1a20 0%, #0a0a0c 100%);
+        }
+        .dashboard-card {
+            background: var(--glass);
+            backdrop-filter: blur(20px);
+            border: 1px solid var(--border);
+            padding: 3rem;
+            border-radius: 24px;
+            text-align: center;
+            max-width: 450px;
+            width: 90%;
+            box-shadow: 0 20px 50px rgba(0,0,0,0.5);
+        }
+        h1 { font-size: 2rem; color: var(--accent); margin-bottom: 0.5rem; }
+        p { color: #888; margin-bottom: 2.5rem; }
+        .key-display {
+            background: rgba(0,0,0,0.3);
+            padding: 1rem;
+            border-radius: 12px;
+            border: 1px dashed var(--accent);
+            font-family: monospace;
+            font-size: 1.2rem;
+            margin-bottom: 2rem;
+            display: none;
+            color: var(--accent);
+        }
+        button {
+            background: var(--accent);
+            color: #fff;
+            border: none;
+            padding: 1rem 2rem;
+            border-radius: 12px;
+            font-weight: 600;
+            font-size: 1rem;
+            cursor: pointer;
+            transition: transform 0.2s, box-shadow 0.2s;
+        }
+        button:hover { transform: translateY(-2px); box-shadow: 0 5px 15px rgba(255,107,0,0.4); }
+        .footer { margin-top: 2rem; font-size: 0.8rem; color: #444; }
+    </style>
+</head>
+<body>
+    <div class="dashboard-card">
+        <h1>MadGames API</h1>
+        <p>Verwalte deine API-Keys für das MadGamesAI Sprachmodell.</p>
+        <div id="key-result" class="key-display"></div>
+        <button onclick="generateKey()">Neuen API-Key erstellen</button>
+        <div class="footer">MadGames AI Engine &bull; Private Space Instance</div>
+    </div>
+    <script>
+        async function generateKey() {
+            const res = await fetch('/generate_key', { method: 'POST' });
+            const data = await res.json();
+            const el = document.getElementById('key-result');
+            el.innerText = data.key;
+            el.style.display = 'block';
+        }
+    </script>
+</body>
+</html>