Update modules/app.py

modules/app.py

@@ -1,13 +1,18 @@
 from fastapi import FastAPI, Request
 from fastapi.responses import HTMLResponse, StreamingResponse
 from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
 from modules.pmbl import PMBL
-.html
+import os
+
+app = FastAPI(docs_url=None, redoc_url=None)
+
+# Mount static files
+app.mount("/static", StaticFiles(directory="static"), name="static")
 
-app = FastAPI()
 pmbl = PMBL("./PMB-7b.Q6_K.gguf")  # Replace with the path to your model
 
-app.mount("/", StaticFiles(directory="static"), name="static")
+templates = Jinja2Templates(directory="static")  # Update the directory to "static"
 
 @app.post("/chat")
 async def chat(request: Request):
@@ -24,7 +29,7 @@ async def chat(request: Request):
 
 @app.get("/", response_class=HTMLResponse)
 async def root(request: Request):
-    return static.TemplateResponse("index.html", {"request": request})
+    return templates.TemplateResponse("index.html", {"request": request})
 
 @app.post("/sleep")
 async def sleep():
@@ -33,4 +38,11 @@ async def sleep():
         return {"message": "Sleep mode completed successfully"}
     except Exception as e:
         print(f"[SYSTEM] Error: {str(e)}")
-        return {"error": str(e)}
+        return {"error": str(e)}
+
+if __name__ == "__main__":
+    import uvicorn
+    import asyncio
+
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 8000))))