Update app.py

app.py

@@ -1,3 +1,5 @@
+import asyncio
+from concurrent.futures import ThreadPoolExecutor
 from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi import FastAPI, HTTPException
@@ -19,12 +21,14 @@ app.add_middleware(
 class GenerateInput(BaseModel):
     prompt: str
 
-def fetch_profile(prompt: str):
+async def fetch_profile(prompt: str):
     if not prompt.strip():
         raise HTTPException(status_code=400, detail="Please provide a non-empty prompt.")
     
-    api = Linkedin('[email protected]', '#845524509suma')
-    c = api.get_profile(prompt)
+    loop = asyncio.get_event_loop()
+    with ThreadPoolExecutor() as pool:
+        api = Linkedin('[email protected]', '#845524509suma')
+        c = await loop.run_in_executor(pool, api.get_profile, prompt)
     return c
 
 @app.head("/")
@@ -32,9 +36,9 @@ async def head_root():
     return {"message": "This is a HEAD request to the root URL."}
 
 @app.get("/", tags=["Home"])
-def api_home():
+async def api_home():
     return {'detail': 'Welcome to FastAPI LinkedIn Crawler by Knowde!'}
 
 @app.post("/api/fetch", tags=["Fetch"])
-def inference(data: GenerateInput):
-    return fetch_profile(data.prompt)
+async def inference(data: GenerateInput):
+    return await fetch_profile(data.prompt)