Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import gradio as gr
 import google.generativeai as genai
 from dotenv import load_dotenv
+import time
 
 # Load environment variables from .env file
 load_dotenv()
@@ -21,12 +22,10 @@ generation_config = {
     "response_mime_type": "text/plain",
 }
 
-# Define safety settings for the model
+# Simplified safety settings (or try removing them to test)
 safety_settings = [
     {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
-    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
-    {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
-    {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
+    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"}
 ]
 
 # Function to generate a response based on user input and chat history
@@ -50,16 +49,22 @@ def generate_response(user_input, chat_history):
     # Limit history length to the last 10 messages
     chat_history = chat_history[-10:]
 
-    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:
-        return f"Error: {str(e)}", chat_history
+    retry_attempts = 3
+    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
+                continue
+            else:
+                return f"Error after {retry_attempts} attempts: {str(e)}", chat_history
 
 # Build the Gradio interface
 with gr.Blocks(theme="Hev832/Applio") as iface: