Update gemini_agent.py

gemini_agent.py

@@ -5,9 +5,10 @@ from excel_parser import ExcelParser
 import re
 import time
 import asyncio
+from google.generativeai.types import HarmCategory, HarmBlockThreshold
 # Add LangChain tools for Wikipedia and DuckDuckGo
-from langchain.tools import DuckDuckGoSearchRun, WikipediaQueryRun
-from langchain.utilities import WikipediaAPIWrapper
+from langchain_community.tools import DuckDuckGoSearchRun, WikipediaQueryRun
+from langchain_community.utilities import WikipediaAPIWrapper
 
 load_dotenv()
 
@@ -78,7 +79,12 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
                     max_output_tokens=50,
                     temperature=0.0
                 ),
-                safety_settings={'HARASSMENT':'block_none', 'HATE_SPEECH':'block_none', 'SEXUALLY_EXPLICIT':'block_none', 'DANGEROUS_CONTENT':'block_none'}
+                safety_settings={
+                    HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
+                    HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
+                    HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
+                    HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
+                }
             )
             answer = response.text.strip()
             
@@ -147,7 +153,12 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
                     max_output_tokens=150,
                     temperature=0.0
                 ),
-                safety_settings={'HARASSMENT':'block_none', 'HATE_SPEECH':'block_none', 'SEXUALLY_EXPLICIT':'block_none', 'DANGEROUS_CONTENT':'block_none'}
+                safety_settings={
+                    HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
+                    HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
+                    HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
+                    HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
+                }
             )
             answer = response.text.strip()
             
@@ -241,7 +252,12 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
                 max_output_tokens=100,
                 temperature=0.0
             ),
-            safety_settings={'HARASSMENT':'block_none', 'HATE_SPEECH':'block_none', 'SEXUALLY_EXPLICIT':'block_none', 'DANGEROUS_CONTENT':'block_none'}
+            safety_settings={
+                HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
+                HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
+                HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
+                HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
+            }
         )
         answer = response.text.strip()
         
@@ -307,7 +323,12 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
                     max_output_tokens=100,
                     temperature=0.0
                 ),
-                safety_settings={'HARASSMENT':'block_none', 'HATE_SPEECH':'block_none', 'SEXUALLY_EXPLICIT':'block_none', 'DANGEROUS_CONTENT':'block_none'}
+                safety_settings={
+                    HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
+                    HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
+                    HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
+                    HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
+                }
             )
             answer = response.text.strip()