Update app.py

app.py

@@ -14,7 +14,7 @@ footer {
 
 load_dotenv()
 
-GEMINI_API_KEY = "AIzaSyBRkv6TCfyZtT5Q6H6ur2W8KSNt9ksbeDI" 
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 
 genai.configure(api_key=GEMINI_API_KEY)
 
@@ -22,20 +22,19 @@ generation_config = {
     "temperature": 0.7,
     "top_p": 0.95,
     "top_k": 64,
-    "max_output_tokens": 512,  # Adjust as needed
+    "max_output_tokens": 512,  
     "response_mime_type": "text/plain",
 }
 
-safety_settings = [
-    {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
-    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"}
-]
-
 mood_prompts = {
     "Fun": "Respond in a light-hearted, playful manner.",
     "Serious": "Respond in a thoughtful, serious tone.",
     "Professional": "Respond in a formal, professional manner.",
-    "Upset": "Respond in a slightly irritated, upset tone."
+    "Upset": "Respond in a slightly irritated, upset tone.",
+    "Empathetic": "Respond in a warm and understanding tone.",
+    "Optimistic": "Respond in a positive, hopeful manner.",
+    "Sarcastic": "Respond with a hint of sarcasm.",
+    "Motivational": "Respond with encouragement and motivation."
 }
 
 def generate_response(user_input, chat_history, mood):
@@ -45,34 +44,27 @@ def generate_response(user_input, chat_history, mood):
     model = genai.GenerativeModel(
         model_name="gemini-1.5-pro", 
         generation_config=generation_config,
-        safety_settings=safety_settings,
         system_instruction=updated_system_content,
     )
 
-    # Add user input to history
     chat_history.append(user_input)
 
-    # Limit history length to the last 10 messages
     chat_history = chat_history[-10:]
 
-    retry_attempts = 3
+    retry_attempts = 10
     for attempt in range(retry_attempts):
         try:
-            # Start a new chat session
             chat_session = model.start_chat()
-
-            # Send the entire chat history as the first message
             response = chat_session.send_message("\n".join(chat_history))
             return response.text, chat_history
 
         except Exception as e:
             if attempt < retry_attempts - 1:
-                time.sleep(2)  # Delay before retrying
+                time.sleep(2)  
                 continue
             else:
                 return f"Error after {retry_attempts} attempts: {str(e)}", chat_history
 
-# Build the Gradio interface
 with gr.Blocks(css=css, theme="bethecloud/storj_theme") as iface:
 
     chat_input = gr.Textbox(lines=2, label="Chatbot", placeholder="Enter your message here...")