Update gemini.py

gemini.py

@@ -1,5 +1,11 @@
-async def complete_gemini_async(infor, key, params={}):
+import requests
+import os
+import httpx
+import asyncio
+
+GEMINI_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent"
 
+async def complete_gemini_async(infor, key, params={}):
     question, sample = infor
     prompt = f"Based on language and the answer of the following sample : {sample} \n Answer this the question : {question}"
     data = {
@@ -23,25 +29,18 @@ async def complete_gemini_async(infor, key, params={}):
     params = {'key': key}
     headers = {"Content-Type": "application/json"}
     result = None
-    pass_data = {
-      "chat": prompt,
-      "key": key,
-      "params": params
-    }
     try:
         async with httpx.AsyncClient() as client:
-            url = rd.choice(gate_geminis)
-            result = await client.post(url, json=pass_data, timeout=120) 
-        result.raise_for_status() 
-        return result.json()
-
+            result = await client.post(GEMINI_URL, params=params, json=data, headers=headers, timeout=60)
+        result.raise_for_status()  
+        result = result.json()["candidates"][0]["content"]["parts"][0]["text"]
+        return {"response":result}
     except requests.RequestException as e:
         if e.status_code == 429:
-            await asyncio.sleep(10)
-
-        print(f"Error making Gemini API request: {e}")
+            await asyncio.sleep(60)
+        #rint(f"Error making Gemini API request: {e}")
         async with httpx.AsyncClient() as client:
-            url = rd.choice(gate_geminis)
-            result = await client.post(url, json=data, timeout=120) 
-        result.raise_for_status() 
-        return result.json()
+            result = await client.post(GEMINI_URL, params=params, json=data, headers=headers, timeout=60)
+        result.raise_for_status()  
+        result = result.json()["candidates"][0]["content"]["parts"][0]["text"]
+        return {"response":result}