New approach part 3

app.py

@@ -120,20 +120,42 @@
 
 
 
-from fastapi import FastAPI
+# from fastapi import FastAPI
+# import httpx
+
+# app = FastAPI()
+# TARGET_SERVER = "http://localhost:11434"
+
+# @app.get("/proxy/{path:path}")
+# async def get_proxy_request(path: str):
+#     async with httpx.AsyncClient() as client:
+#         response = await client.get(f"{TARGET_SERVER}/{path}")
+#     return response.json()
+
+# @app.post("/proxy/{path:path}")
+# async def post_proxy_request(path: str):
+#     async with httpx.AsyncClient() as client:
+#         response = await client.post(f"{TARGET_SERVER}/{path}")
+#     return response.json()
+
+
+from fastapi import FastAPI, Request
 import httpx
 
 app = FastAPI()
 TARGET_SERVER = "http://localhost:11434"
 
-@app.get("/proxy/{path:path}")
-async def get_proxy_request(path: str):
+@app.get("/{path:path}")
+async def proxy_get(path: str):
+    """ Forwards GET requests to the target server """
     async with httpx.AsyncClient() as client:
         response = await client.get(f"{TARGET_SERVER}/{path}")
     return response.json()
 
-@app.post("/proxy/{path:path}")
-async def post_proxy_request(path: str):
+@app.post("/{path:path}")
+async def proxy_post(path: str, request: Request):
+    """ Forwards POST requests to the target server """
+    data = await request.json()  # Get the request body
     async with httpx.AsyncClient() as client:
-        response = await client.post(f"{TARGET_SERVER}/{path}")
+        response = await client.post(f"{TARGET_SERVER}/{path}", json=data)
     return response.json()