Spaces:

Raiff1982
/

codette-ai

Running

App Files Files Community

Raiff1982 commited on 21 days ago

Commit

ff09025

verified ·

1 Parent(s): e801c72

Update app.py

Browse files

Files changed (1) hide show

app.py +156 -7

app.py CHANGED Viewed

@@ -16,7 +16,8 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 if not HF_TOKEN:
     raise RuntimeError("HF_TOKEN missing. Set in HF Space secrets.")
-client = InferenceClient(token=HF_TOKEN)
 # ─────────────────────────────────────────────
 # FASTAPI INIT
@@ -32,13 +33,162 @@ app.add_middleware(
 )
 # ─────────────────────────────────────────────
-# ROOT (SERVES FRONTEND)
 # ─────────────────────────────────────────────
 @app.get("/", response_class=HTMLResponse)
 async def root():
-    with open("index.html", encoding="utf-8") as f:
-        return f.read()
 # ─────────────────────────────────────────────
 # DEBUG
@@ -57,7 +207,7 @@ async def test():
         return {"status": "error", "error": str(e)}
 # ─────────────────────────────────────────────
-# CHAT ENDPOINT
 # ─────────────────────────────────────────────
 @app.post("/api/chat")
@@ -110,11 +260,10 @@ async def chat(request: Request):
     return StreamingResponse(event_stream(), media_type="application/x-ndjson")
 # ─────────────────────────────────────────────
 # RUN
 # ─────────────────────────────────────────────
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)

 if not HF_TOKEN:
     raise RuntimeError("HF_TOKEN missing. Set in HF Space secrets.")
+# ✅ FIX: bind model correctly (this was your silent bug)
+client = InferenceClient(model=MODEL_ID, token=HF_TOKEN)
 # ─────────────────────────────────────────────
 # FASTAPI INIT
 )
 # ─────────────────────────────────────────────
+# ROOT (CODETTE UI)
 # ─────────────────────────────────────────────
 @app.get("/", response_class=HTMLResponse)
 async def root():
+    return """
+<!DOCTYPE html>
+<html>
+<head>
+<title>Codette</title>
+<style>
+body {
+    margin:0;
+    font-family: Inter, sans-serif;
+    background: radial-gradient(circle at top, #14142b, #0b0b17);
+    color:#e5e7eb;
+}
+.container {
+    display:flex;
+    height:100vh;
+}
+.left {
+    flex:3;
+    display:flex;
+    flex-direction:column;
+    padding:15px;
+}
+.right {
+    flex:1;
+    padding:15px;
+}
+.chat {
+    flex:1;
+    overflow:auto;
+    background: rgba(20,20,40,0.7);
+    border-radius:10px;
+    padding:10px;
+}
+.input {
+    display:flex;
+    margin-top:10px;
+}
+input {
+    flex:1;
+    padding:10px;
+    background:#0f0f1e;
+    border:1px solid #06b6d4;
+    color:white;
+}
+button {
+    padding:10px;
+    background: linear-gradient(135deg,#a855f7,#06b6d4);
+    border:none;
+    color:white;
+    cursor:pointer;
+}
+.metric {
+    background: rgba(20,20,40,0.7);
+    padding:10px;
+    margin-bottom:10px;
+    border-radius:8px;
+    font-family: monospace;
+}
+</style>
+</head>
+<body>
+<div class="container">
+<div class="left">
+    <h2>CODETTE</h2>
+    <div id="chat" class="chat"></div>
+    <div class="input">
+        <input id="msg" placeholder="Ask Codette..." />
+        <button onclick="send()">▶</button>
+    </div>
+</div>
+<div class="right">
+    <div class="metric" id="metrics">
+    Γ: 0.0000<br>
+    η: 0.0000<br>
+    Risk: LOW
+    </div>
+</div>
+</div>
+<script>
+async function send() {
+    const input = document.getElementById("msg");
+    const chat = document.getElementById("chat");
+    const userText = input.value;
+    if (!userText) return;
+    chat.innerHTML += "<div><b>You:</b> " + userText + "</div>";
+    input.value = "";
+    const res = await fetch("/api/chat", {
+        method:"POST",
+        headers:{"Content-Type":"application/json"},
+        body: JSON.stringify({
+            messages:[{role:"user", content:userText}]
+        })
+    });
+    const reader = res.body.getReader();
+    const decoder = new TextDecoder();
+    let botDiv = document.createElement("div");
+    botDiv.innerHTML = "<b>Codette:</b> ";
+    chat.appendChild(botDiv);
+    while (true) {
+        const {done, value} = await reader.read();
+        if (done) break;
+        const chunk = decoder.decode(value);
+        const lines = chunk.split("\\n");
+        for (let line of lines) {
+            if (!line.trim()) continue;
+            try {
+                const data = JSON.parse(line);
+                if (data.message?.content) {
+                    botDiv.innerHTML += data.message.content;
+                }
+            } catch {}
+        }
+    }
+    chat.scrollTop = chat.scrollHeight;
+    // fake metrics update (you can wire real later)
+    document.getElementById("metrics").innerHTML =
+        "Γ: " + (0.8 + Math.random()*0.2).toFixed(4) + "<br>" +
+        "η: " + (0.6 + Math.random()*0.3).toFixed(4) + "<br>" +
+        "Risk: LOW";
+}
+</script>
+</body>
+</html>
+"""
 # ─────────────────────────────────────────────
 # DEBUG
         return {"status": "error", "error": str(e)}
 # ─────────────────────────────────────────────
+# CHAT ENDPOINT (UNCHANGED CORE)
 # ─────────────────────────────────────────────
 @app.post("/api/chat")
     return StreamingResponse(event_stream(), media_type="application/x-ndjson")
 # ─────────────────────────────────────────────
 # RUN
 # ─────────────────────────────────────────────
 if __name__ == "__main__":
     import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)