Update gemini.py

gemini.py

@@ -1,13 +1,17 @@
-import requests
-import os
-import httpx
-import asyncio
+async def complete_gemini_async(infor, key, params={}):
 
-GEMINI_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent"
-
-async def complete_gemini_async(chat, 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 = {
-        "contents": chat,
+        "contents":  [
+            {
+                "parts": [
+                    {
+                        "text":prompt
+                    }
+                ]
+            }
+        ],
         "generationConfig": {
             "stopSequences": ["Title"],
             "temperature": 1.0 if "temp" not in params else params["temp"],
@@ -19,18 +23,25 @@ async def complete_gemini_async(chat, 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:
-            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}
+            url = rd.choice(gate_geminis)
+            result = await client.post(url, json=pass_data, timeout=120) 
+        result.raise_for_status() 
+        return result.json()
+
     except requests.RequestException as e:
         if e.status_code == 429:
-            await asyncio.sleep(60)
-        #rint(f"Error making Gemini API request: {e}")
+            await asyncio.sleep(10)
+
+        print(f"Error making Gemini API request: {e}")
         async with httpx.AsyncClient() as client:
-            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}
+            url = rd.choice(gate_geminis)
+            result = await client.post(url, json=data, timeout=120) 
+        result.raise_for_status() 
+        return result.json()