Update app.py

app.py

@@ -6,10 +6,9 @@ import traceback
 from fastapi.middleware.cors import CORSMiddleware
 from gemini import complete_gemini_async
 
-class GenItem(BaseModel):
-    chat:list
-    key:str
-    params:dict
+class Text(BaseModel):
+    ques: str
+    ans: str
     
 app = FastAPI()
 origins = ["*"]
@@ -20,10 +19,10 @@ app.add_middleware(
     allow_methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"],
     allow_headers=["*"],
 )
-@app.post("/complete")
-async def complete(item: GenItem):
+@app.post("/geminiv2")
+async def geminiv2(data: Text, api_key: str = Security(get_api_key)):
     try:
-        result = await complete_gemini_async(item.chat, item.key, item.params)
+        result = await complete_gemini_async((data.ques, data.ans), "ahihi")
         return result
     except Exception as e:
         raise HTTPException(status_code=500, detail="An error occurred: {}".format(traceback.format_exc()))