Update app.py

app.py

@@ -1,9 +1,8 @@
 import torch
-from fastapi import FastAPI, Request, Form
-from fastapi.responses import HTMLResponse, JSONResponse
-from fastapi.staticfiles import StaticFiles
-from fastapi.templating import Jinja2Templates
+from fastapi import FastAPI, Form
+from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import json
 
 # ---------------------------
 # Models
@@ -11,7 +10,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 MODEL_OPTIONS = {
     "DeepSeek Coder 1.3B": "deepseek-ai/deepseek-coder-1.3b-instruct",
     "StarCoder 1B": "bigcode/starcoderbase-1b",
-    "CodeLLaMA 7B": "codellama/CodeLlama-7b-Instruct-hf"
+    "CodeLLaMA 7B": "codellama/CodeLLaMA-7b-Instruct-hf"
 }
 loaded_models = {}
 
@@ -31,19 +30,22 @@ def get_model(model_key):
 # FastAPI setup
 # ---------------------------
 app = FastAPI()
-app.mount("/static", StaticFiles(directory="static"), name="static")
-templates = Jinja2Templates(directory=".")
 
 # ---------------------------
 # Routes
 # ---------------------------
 @app.get("/", response_class=HTMLResponse)
-async def home(request: Request):
-    return templates.TemplateResponse("index.html", {"request": request, "models": list(MODEL_OPTIONS.keys())})
+async def home():
+    # Just serve index.html directly from root
+    return FileResponse("index.html")
+
+@app.get("/style.css")
+async def css():
+    # Serve CSS directly
+    return FileResponse("style.css")
 
 @app.post("/chat")
 async def chat(user_input: str = Form(...), model_choice: str = Form(...), history: str = Form("[]")):
-    import json
     history = json.loads(history)
 
     tokenizer, model = get_model(model_choice)