Update main.py

main.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from duckai import DuckAI
+from g4f.client import Client
 
 app = FastAPI()
 
@@ -22,9 +23,15 @@ async def chat(query: str):
     if not query:
         raise HTTPException(status_code=400, detail="Query parameter is required")
 
-    duck = DuckAI()
+    client = Client()
+    duck = DuckAI()  # Initialize DuckAI here
+    
     try:
-        results = duck.chat(query, model='gpt-4o-mini')
+        results = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[{"role": "user", "content": query}],
+            web_search=False
+        )
         return {"results": results}
     except Exception as e1:
         print(f"Primary model (gpt-4o-mini) failed: {e1}")
@@ -42,19 +49,11 @@ async def chat(query: str):
                 }
             )
 
-async def chat_with_model(query: str, model: str):
-    try:
-        duck = DuckAI()
-        results = duck.chat(query, model=model)
-        return {"results": results}
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
 @app.get("/health")
 @app.get("/")
 async def health_check():
-    
     return {"status": "healthy"}
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info", reload=True)
+    uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info", reload=True)