Update gemini_agent.py

gemini_agent.py

@@ -3,6 +3,8 @@ import google.generativeai as genai
 from dotenv import load_dotenv
 from excel_parser import ExcelParser
 import re
+import time
+import asyncio
 
 load_dotenv()
 
@@ -15,6 +17,8 @@ class GeminiAgent:
         genai.configure(api_key=api_key)
         
         self.model = genai.GenerativeModel('gemini-2.0-flash-exp')
+        self.last_request_time = 0
+        self.min_request_interval = 1.0  # 1 second between requests
         
         # Initialize parsers
         self.excel_parser = ExcelParser()
@@ -61,6 +65,7 @@ class GeminiAgent:
 Provide only the direct answer. If it's a quote, give just the quoted text. If it's a number, give just the number. If it's about bird species count, analyze carefully and give the exact count. If it's about dialogue, provide the exact words spoken."""
         
         try:
+            await self._rate_limit()
             response = self.model.generate_content(
                 video_prompt,
                 generation_config=genai.types.GenerationConfig(
@@ -128,6 +133,7 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
         If you don't have specific information about this Excel file, provide a reasonable estimate based on similar data."""
         
         try:
+            await self._rate_limit()
             response = self.model.generate_content(
                 excel_prompt,
                 generation_config=genai.types.GenerationConfig(
@@ -213,6 +219,7 @@ Answer:"""
         
         # Use the constructed prompt for all cases
         
+        await self._rate_limit()
         response = self.model.generate_content(
             prompt,
             generation_config=genai.types.GenerationConfig(
@@ -252,6 +259,7 @@ Answer:"""
         Provide only the direct answer without explanation."""
         
         try:
+            await self._rate_limit()
             response = self.model.generate_content(
                 prompt,
                 generation_config=genai.types.GenerationConfig(
@@ -270,4 +278,12 @@ Answer:"""
             
         except Exception as e:
             print(f"Failed to generate video answer: {str(e)}")
-            return "Video analysis unavailable."
+            return "Video analysis unavailable."
+    
+    async def _rate_limit(self):
+        """Ensure minimum time between API requests"""
+        current_time = time.time()
+        time_since_last = current_time - self.last_request_time
+        if time_since_last < self.min_request_interval:
+            await asyncio.sleep(self.min_request_interval - time_since_last)
+        self.last_request_time = time.time()