Update main.py

main.py

@@ -177,6 +177,17 @@ async def get_completion(payload: Payload):
             raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {e}")
 
     return StreamingResponse(stream_generator(payload_dict), media_type="application/json")
+@app.get("/playground", response_class=HTMLResponse)
+async def playground():
+    # Open and read the content of playground.html (in the same folder as the app)
+    file_path = "playground.html"
+
+    try:
+        with open(file_path, "r") as file:
+            html_content = file.read()
+        return HTMLResponse(content=html_content)
+    except FileNotFoundError:
+        return HTMLResponse(content="<h1>playground.html not found</h1>", status_code=404)
 @app.on_event("startup")
 async def startup_event():
     print("API endpoints:")