Update app.py

app.py

@@ -6,13 +6,11 @@ from pypdf import PdfReader
 import gradio as gr
 import csv
 
-print("hi")
-
 # Load environment variables
 load_dotenv(override=True)
 
 GEMINI_BASE_URL = "https://generativelanguage.googleapis.com/v1beta/openai/"
-google_api_key = os.getenv("GOOGLE_API_KEY")
+google_api_key = os.getenv("GOOGLE_API_KEY","AIzaSyDGXQliok_IEo51FGx_7EWiJ_WlCj0e0qc")
 gemini = OpenAI(base_url=GEMINI_BASE_URL, api_key=google_api_key)
 
 # CSV files for logging
@@ -150,40 +148,36 @@ class Me:
 
     # Main chat function
     def chat(self, message, history):
-        # 1. Correctly format the messages list from Gradio's history tuples.
-        messages = [{"role": "system", "content": self.system_prompt()}]
-        for user_msg, assistant_msg in history:
-            messages.append({"role": "user", "content": user_msg})
-            messages.append({"role": "assistant", "content": assistant_msg})
-        messages.append({"role": "user", "content": message})
-
-        # 2. First API call: The model decides if it needs a tool.
+        messages = [{"role": "system", "content": self.system_prompt()}] + history + [{"role": "user", "content": message}]
+        done = False
+        #while not done:
+        # Send the first message
         response = self.openai.chat.completions.create(
             model="gemini-2.5-flash-preview-05-20",
             messages=messages,
             tools=tools
         )
-
-        # 3. Check for a tool call.
+    
+        # Check if the response contains tool calls
         if response.choices[0].finish_reason == "tool_calls":
-            message_with_tool_calls = response.choices[0].message
-            tool_calls = message_with_tool_calls.tool_calls
+            message_to_add = response.choices[0].message
+            tool_calls_to_execute = message_to_add.tool_calls
 
-            # 4. Execute the tool and get the result.
-            results = self.handle_tool_call(tool_calls)
+            # Execute the tool and get the result.
+            results = self.handle_tool_call(tool_calls_to_execute)
 
-            # 5. Append the tool call and its output to the messages list.
-            messages.append(message_with_tool_calls)
+            # Append the tool call and its output to the messages list.
+            messages.append(message_to_add)
             messages.extend(results)
-
-            # 6. Second API call: Get the final, conversational response.
-            final_response = self.openai.chat.completions.create(
+            
+            # Now call the API again with the tool output
+            response = self.openai.chat.completions.create(
                 model="gemini-2.5-flash-preview-05-20",
-                messages=messages
+                messages=messages,
             )
-            return final_response.choices[0].message.content
+            return response.choices[0].message.content
         else:
-            # If no tool call, return the direct response.
+            # If no tool call, return the direct response
             return response.choices[0].message.content