Update app.py

app.py

@@ -82,9 +82,15 @@ class Me:
         #     print(f"Checking Keys: Open router API Key exists and begins {open_router_api_key[:8]}")
         # else:
         #     print("Checking Keys: Open router API Key not set - please head to the troubleshooting guide in the setup folder")
-        self.openrouter = OpenAI(
-                            base_url="https://openrouter.ai/api/v1",
-                            api_key= os.getenv('OPEN_ROUTER_API_KEY') )  # open_router_api_key
+        # self.openrouter = OpenAI(
+        #                     base_url="https://openrouter.ai/api/v1",
+        #                     api_key= os.getenv('OPEN_ROUTER_API_KEY') )  # open_router_api_key
+
+        # Initialize Gemini client using OpenAI format
+        self.gemini = OpenAI(
+            api_key=os.getenv("GOOGLE_API_KEY"), 
+            base_url="https://generativelanguage.googleapis.com/v1beta/openai/"
+        )
        
         self.name = "Chaoran Zhou"
         reader = PdfReader("me/linkedin.pdf")
@@ -128,7 +134,14 @@ If the user is engaging in discussion, try to steer them towards getting in touc
         
         # Generate initial response with tool handling
         while not done:
-            response = self.openrouter.chat.completions.create(model="meta-llama/llama-3.3-8b-instruct:free", messages=messages, tools=tools)
+            # response = self.openrouter.chat.completions.create(model="meta-llama/llama-3.3-8b-instruct:free", messages=messages, tools=tools)
+            
+            response = self.gemini.chat.completions.create(
+                model="gemini-2.5-flash-preview-05-20",
+                messages=messages, 
+                tools=tools
+            )
+            
             if response.choices[0].finish_reason=="tool_calls":
                 message = response.choices[0].message
                 tool_calls = message.tool_calls