Update app.py

app.py

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