Update ff.py

ff.py

@@ -4,21 +4,37 @@ import math as matha
 from g4f.Provider import (
     Bard) 
 import g4f
+from llama_index.llms.gemini import Gemini
 #from bardapi import Bard as Bd
 import os 
 token=os.environ.get("TOKEN")
 toto = os.environ.get("BARD") 
-
+safe = [
+    {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "threshold": "BLOCK_NONE",
+    },
+]
 
 #bard = Bd(token=toto)
-import google.generativeai as genai
-
-genai.configure(
-    api_key=token
-)
-model = genai.GenerativeModel(
-    model_name='gemini-pro'
-)
+
+
+os.environ["GOOGLE_API_KEY"] = token
+
+llm = Gemini(model="models/gemini-pro") 
+
 model_vision = genai.GenerativeModel('gemini-pro-vision')
 from g4f import Provider, models
 from langchain.llms.base import LLM
@@ -46,15 +62,14 @@ def rees(input_text):
     return re
 
 
-def res(input_text):
-    response = g4f.ChatCompletion.create(
-    model=g4f.models.gpt_35_turbo,
-    messages=[{"role": "user", "content":input_text}],
-)  # Alternative model setting
+async def greet(tt):
+    response = await llm.acomplete(tt,safety_settings=safe)
+    print(response)
     return response 
 
 
 
+
 def infer(im):
     im.save("converted.png")
     url = "https://ajax.thehive.ai/api/demo/classify?endpoint=text_recognition"
@@ -92,9 +107,8 @@ def gpt(prompt):
     #answer = res(prompt)
     
     print(prompt)
-    response = model.generate_content(prompt)
-    print(response.text)
-    return response.text
+    response = greet(prompt) 
+    return response
 
 def gpt_francais(french_prompt,choix,autheur):
     
@@ -193,8 +207,8 @@ CONCLUSION
     """answer = rees(haha)
     print(answer)
     return answer"""
-    answer = model.generate_content(haha)
-    return answer.text
+    answer = greet(haha)
+    return answer
 
 
 def gpt_hist(hist_prompt,p_1,p_2,p_3):
@@ -218,9 +232,9 @@ CONCLUSION
 2. question douverture du sujet.   
     """
     
-    answer = model.generate_content(histt_prompt)
+    answer = greet(histt_prompt)
     print(answer)
-    return answer.text
+    return answer