Update app.py

app.py

@@ -3,7 +3,7 @@ from fastapi.responses import StreamingResponse
 import uvicorn
 from v1 import v1
 from v2 import v2
-
+from chatv1 import CHATv1
 app = FastAPI()
 
 @app.get("/Search/pro")
@@ -28,6 +28,26 @@ async def v2_chat(prompt: str):
             yield f"data: {chunk}\n\n"
     
     return StreamingResponse(response_generator(), media_type="text/event-stream")
+@app.get("/chatv1")
+async def chat_endpoint_get(
+    user_prompt: str = Query(..., description="User's prompt"),
+    system_prompt: Optional[str] = Query("You are a helpful AI assistant.", description="System prompt to set AI behavior")
+):
+    def generate():
+        ai = CHATv1()
+        for chunk in ai.chat(user_prompt, system_prompt):
+            yield f"data: {chunk}\n\n"
+    
+    return StreamingResponse(generate(), media_type="text/event-stream")
+
+@app.post("/chatv1")
+async def chat_endpoint_post(request: ChatRequest):
+    def generate():
+        ai = CHATv1()
+        for chunk in ai.chat(request.user_prompt, request.system_prompt):
+            yield f"data: {chunk}\n\n"
+    
+    return StreamingResponse(generate(), media_type="text/event-stream")
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=8000)