Update chatbot_handler.py

chatbot_handler.py

@@ -10,8 +10,7 @@ import asyncio
 GEMINI_API_KEY = os.getenv('GEMINI_API_KEY', '')
 
 client = None
-model_name = "gemini-1.5-flash-latest" # Using a more recent Flash model
-# model_name = "gemini-2.0-flash" # As per user's documentation snippet, ensure this model is available with their API key type
+model_name = "gemini-2.0-flash" # As per user's documentation snippet, ensure this model is available with their API key type
 
 # This will be used to create genai_types.GenerateContentConfig
 generation_config_params = {
@@ -23,13 +22,6 @@ generation_config_params = {
     # "system_instruction": "You are a helpful AI assistant providing insights on LinkedIn analytics."
 }
 
-# Safety settings list
-common_safety_settings = [
-    {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
-    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
-    {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
-    {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
-]
 
 try:
     if GEMINI_API_KEY:
@@ -98,8 +90,8 @@ async def generate_llm_response(user_message: str, plot_id: str, plot_label: str
                 client.models.generate_content,
                 model=model_name,  # Use model_name directly
                 contents=gemini_formatted_history,
-                config=gen_config_obj,  # Corrected keyword argument
-                safety_settings=common_safety_settings
+                config=gen_config_obj  # Corrected keyword argument
+                
             )
         else:
             logging.error(f"Gemini client (genai.Client) does not have 'models.generate_content' method. Type: {type(client)}")