Update app.py

app.py

@@ -3,17 +3,13 @@ import os
 import google.generativeai as genai
 import logging
 import time
-import backoff
+#import backoff
 
 # Configure Logging
 logging.basicConfig(level=logging.ERROR, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # Load environment variables
-try:
-    genai.configure(api_key=os.environ["geminiapikey"])
-except KeyError:
-    logging.error("Error: 'geminiapikey' environment variable not found.")
-    exit(1)
+genai.configure(api_key=os.environ["geminiapikey"])
 
 read_key = os.environ.get('HF_TOKEN', None)
 
@@ -28,13 +24,11 @@ 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 = {
-        "temperature": 0.7,
+        "temperature": 0.4,
         "top_p": 0.95,
         "top_k": 40,
         "max_output_tokens": 2048,
@@ -46,23 +40,9 @@ def predict(prompt):
         generation_config=generation_config,
     )
 
-    try:
-        contents_to_send = [genai.Content(parts=[prompt])]
-
-        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}"
-
+    contents_to_send = [genai.Content(parts=[prompt])]       
+    response = model.generate_content(contents=contents_to_send, tools='google_search_retrieval')
+    return response.text
 
 # Create the Gradio interface
 with gr.Blocks(css=custom_css) as demo: