Update app.py

app.py

@@ -9,8 +9,8 @@ import gradio as gr
 
 load_dotenv(override=True)
 google_api_key = os.getenv('GOOGLE_API_KEY')
-gemini = OpenAI(api_key=google_api_key, base_url="https://generativelanguage.googleapis.com/v1beta/openai/")
-model_name = "gemini-2.0-flash"
+#gemini = OpenAI(api_key=google_api_key, base_url="https://generativelanguage.googleapis.com/v1beta/openai/")
+#model_name = "gemini-2.0-flash"
 
 def push(text):
     requests.post(
@@ -79,8 +79,10 @@ tools = [{"type": "function", "function": record_user_details_json},
 class Me:
 
     def __init__(self):
-        self.openai = OpenAI()
+        #self.openai = OpenAI()
+        self.gemini = OpenAI(api_key=google_api_key, base_url="https://generativelanguage.googleapis.com/v1beta/openai/")
         self.name = "Ed Donner"
+        self.model_name = "gemini-2.0-flash"
         reader = PdfReader("me/linkedin.pdf")
         self.linkedin = ""
         for page in reader.pages:
@@ -120,7 +122,7 @@ If the user is engaging in discussion, try to steer them towards getting in touc
         done = False
         while not done:
             #response = self.openai.chat.completions.create(model="gpt-4o-mini", messages=messages, tools=tools)
-            response = gemini.chat.completions.create(model=model_name, messages=messages,  tools=tools)
+            response = self.gemini.chat.completions.create(model=self.model_name, messages=messages,  tools=tools)
             if response.choices[0].finish_reason=="tool_calls":
                 message = response.choices[0].message
                 tool_calls = message.tool_calls