ARCH.AI_SPACE

Sleeping

App Files Files Community

Gemini commited on Oct 1, 2025

Commit

39dcb26

1 Parent(s): d078b16

feat: Add chat interface and versioned API

Browse files

Files changed (5) hide show

Dockerfile +3 -1
chat.html +21 -0
main.py +47 -4
static/script.js +47 -0
static/style.css +68 -0

Dockerfile CHANGED Viewed

@@ -16,10 +16,12 @@ RUN apt-get update && apt-get install -y python3 python3-pip curl
 WORKDIR /app
 COPY ./main.py /app/main.py
 COPY ./start.sh /app/start.sh
 RUN chmod +x /app/start.sh
 # 3. Install Python dependencies for the FastAPI gateway
-RUN pip3 install fastapi uvicorn requests python-dotenv --break-system-packages
 # Expose the port the FastAPI gateway will listen on
 EXPOSE 7860

 WORKDIR /app
 COPY ./main.py /app/main.py
 COPY ./start.sh /app/start.sh
+COPY ./chat.html /app/chat.html
+COPY ./static /app/static
 RUN chmod +x /app/start.sh
 # 3. Install Python dependencies for the FastAPI gateway
+RUN pip3 install fastapi uvicorn requests --break-system-packages
 # Expose the port the FastAPI gateway will listen on
 EXPOSE 7860

chat.html ADDED Viewed

	@@ -0,0 +1,21 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Ollama Chat</title>
+    <link rel="stylesheet" href="/static/style.css">
+</head>
+<body>
+    <div id="chat-container">
+        <div id="chat-window">
+            <ul id="message-list"></ul>
+        </div>
+        <div id="input-container">
+            <input type="text" id="message-input" placeholder="Type your message...">
+            <button id="send-button">Send</button>
+        </div>
+    </div>
+    <script src="/static/script.js"></script>
+</body>
+</html>

main.py CHANGED Viewed

@@ -1,7 +1,8 @@
 import os
 import requests
-from fastapi import FastAPI, Request, HTTPException
-from fastapi.responses import JSONResponse
 # Get the secret API key from the environment variables (set in Space secrets)
 # This is the key the user must provide to access the service.
@@ -12,8 +13,14 @@ OLLAMA_API_URL = "http://localhost:11434"
 app = FastAPI()
-@app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE"])
-async def proxy(request: Request, path: str):
     # 1. Check for the API Key
     provided_key = request.headers.get("X-API-Key")
     if not AUTH_KEY or provided_key != AUTH_KEY:
@@ -42,3 +49,39 @@ async def proxy(request: Request, path: str):
         )
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))

 import os
 import requests
+from fastapi import FastAPI, Request, HTTPException, APIRouter
+from fastapi.responses import JSONResponse, RedirectResponse, FileResponse
+from fastapi.staticfiles import StaticFiles
 # Get the secret API key from the environment variables (set in Space secrets)
 # This is the key the user must provide to access the service.
 app = FastAPI()
+# Mount the static directory to serve CSS and JS files
+app.mount("/static", StaticFiles(directory="static"), name="static")
+# API v1 Router (secured)
+api_v1 = APIRouter()
+@api_v1.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE"])
+async def proxy_v1(request: Request, path: str):
     # 1. Check for the API Key
     provided_key = request.headers.get("X-API-Key")
     if not AUTH_KEY or provided_key != AUTH_KEY:
         )
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
+app.include_router(api_v1, prefix="/api/v1")
+# Chat API endpoint (not secured)
+@app.post("/api/chat")
+async def chat_endpoint(request: Request):
+    body = await request.json()
+    model = body.get("model", "llama3")
+    prompt = body.get("prompt")
+    if not prompt:
+        raise HTTPException(status_code=400, detail="Prompt is required")
+    url = f"{OLLAMA_API_URL}/api/generate"
+    try:
+        response = requests.post(
+            url=url,
+            json={"model": model, "prompt": prompt, "stream": False}
+        )
+        return JSONResponse(
+            status_code=response.status_code,
+            content=response.json()
+        )
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+@app.get("/")
+async def root():
+    return RedirectResponse(url="/chat")
+@app.get("/chat")
+async def chat_page():
+    return FileResponse('chat.html')

static/script.js ADDED Viewed

	@@ -0,0 +1,47 @@

+const messageList = document.getElementById('message-list');
+const messageInput = document.getElementById('message-input');
+const sendButton = document.getElementById('send-button');
+sendButton.addEventListener('click', sendMessage);
+messageInput.addEventListener('keydown', (event) => {
+    if (event.key === 'Enter') {
+        sendMessage();
+    }
+});
+function sendMessage() {
+    const message = messageInput.value.trim();
+    if (message === '') {
+        return;
+    }
+    appendMessage(message, 'user-message');
+    messageInput.value = '';
+    fetch('/api/chat', {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json'
+        },
+        body: JSON.stringify({
+            model: 'llama3', // You can change the model here
+            prompt: message
+        })
+    })
+    .then(response => response.json())
+    .then(data => {
+        appendMessage(data.response, 'bot-message');
+    })
+    .catch(error => {
+        console.error('Error:', error);
+        appendMessage('Sorry, something went wrong.', 'bot-message');
+    });
+}
+function appendMessage(message, className) {
+    const li = document.createElement('li');
+    li.textContent = message;
+    li.classList.add(className);
+    messageList.appendChild(li);
+    messageList.scrollTop = messageList.scrollHeight;
+}

static/style.css ADDED Viewed

	@@ -0,0 +1,68 @@

+body {
+    font-family: sans-serif;
+    margin: 0;
+    background-color: #f0f0f0;
+}
+#chat-container {
+    display: flex;
+    flex-direction: column;
+    height: 100vh;
+    max-width: 800px;
+    margin: 0 auto;
+    background-color: white;
+    border-left: 1px solid #ccc;
+    border-right: 1px solid #ccc;
+}
+#chat-window {
+    flex-grow: 1;
+    overflow-y: auto;
+    padding: 20px;
+}
+#message-list {
+    list-style: none;
+    margin: 0;
+    padding: 0;
+}
+#message-list li {
+    margin-bottom: 10px;
+    padding: 10px;
+    border-radius: 5px;
+}
+#message-list .user-message {
+    background-color: #dcf8c6;
+    align-self: flex-end;
+}
+#message-list .bot-message {
+    background-color: #f1f0f0;
+    align-self: flex-start;
+}
+#input-container {
+    display: flex;
+    padding: 20px;
+    border-top: 1px solid #ccc;
+}
+#message-input {
+    flex-grow: 1;
+    border: 1px solid #ccc;
+    border-radius: 5px;
+    padding: 10px;
+    font-size: 16px;
+}
+#send-button {
+    background-color: #4CAF50;
+    color: white;
+    border: none;
+    padding: 10px 20px;
+    border-radius: 5px;
+    margin-left: 10px;
+    cursor: pointer;
+}