Update app.py

app.py

@@ -71,6 +71,7 @@ class Me:
         return prompt
 
     def chat(self, message, history):
+        # This is the corrected history management logic.
         if not history:
             messages = [{"role": "user", "parts": [self.system_prompt()]}]
         else:
@@ -78,13 +79,18 @@ class Me:
             for user_msg, model_msg in history:
                 messages.append({"role": "user", "parts": [user_msg]})
                 messages.append({"role": "model", "parts": [model_msg]})
+
         messages.append({"role": "user", "parts": [message]})
-        
+
         done = False
         while not done:
-            response = self.model.generate_content(messages, tools=tools)
+            response = self.model.generate_content(
+                messages,
+                tools=tools,
+            )
             candidate = response.candidates[0]
             finish_reason = candidate.finish_reason
+
             if finish_reason == "TOOL_USE":
                 tool_call = candidate.content.parts[0].function_call
                 result = self.handle_tool_call(tool_call)
@@ -92,6 +98,7 @@ class Me:
                 messages.append(result)
             else:
                 done = True
+
         return candidate.content.parts[0].text
 
 if __name__ == "__main__":