Update app.py

app.py

@@ -10,8 +10,23 @@ genai.configure(api_key=GOOGLE_AI_STUDIO)
 model = genai.GenerativeModel('gemini-pro')
 
 def get_response(query):
-    response = model.generate_content(query)
-    return response.text
+    """
+    Searches for content based on the provided query using the Gemini model.
+    Handles DeadlineExceeded exceptions from the Google API.
+    Args:
+        query (str): The search query.
+    Returns:
+        str: The response text from the Gemini model or an error message.
+    """
+    try:
+        response = model.generate_content(query)
+        return response.text
+    except exceptions.DeadlineExceeded as e:
+        # Handle the DeadlineExceeded exception here
+        print("Error: Deadline Exceeded -", str(e))
+        # You can return a custom message or take other appropriate actions
+        return "Error: The request timed out. Please try again later."
+
 
 # Gradio interface
 iface = gr.Interface(