Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import os
 import google.generativeai as genai
 import logging
+import time
+import backoff
 
 # Configure Logging
 logging.basicConfig(level=logging.ERROR, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -26,6 +28,9 @@ custom_css = """
 }
 """
 
+@backoff.on_exception(backoff.expo,
+                      (genai.APIError),
+                      max_tries=3) # retry up to 3 times
 def predict(prompt):
     # Create the model
     generation_config = {
@@ -42,18 +47,21 @@ def predict(prompt):
     )
 
     try:
-        #contents_to_send = [genai.Content(parts=[prompt])]
+        contents_to_send = [genai.Content(parts=[prompt])]
 
-        response = model.generate_content(contents=prompt, tools='google_search_retrieval')
+        response = model.generate_content(contents=contents_to_send, tools='google_search_retrieval')
 
         if response and response.text:
           return response.text
         else:
           logging.error(f"Unexpected response: {response}")
           return "Error: Could not extract text from the response."
+    except genai.APIError as e:
+       logging.error(f"API error occurred: {e}")
+       raise
     except Exception as e:
-      logging.error(f"An error occurred: {e}")
-      return f"An error occurred: {e}"
+        logging.error(f"An error occurred: {e}")
+        return f"An error occurred: {e}"
 
 
 # Create the Gradio interface