Update app.py

app.py

@@ -14,8 +14,15 @@ genai.configure(api_key=GOOGLE_AI_STUDIO)
 model = genai.GenerativeModel('gemini-pro')
 
 def generate_response(query):
-    response = model.generate_content(query)
-    return response.text
+    try:
+        response = model.generate_content(query)
+        return response.text
+    except exceptions.DeadlineExceeded as e:
+        print("Deadline Exceeded Error: ", e)
+        return "Error: Deadline Exceeded"
+    except Exception as e:
+        print("An unexpected error occurred: ", e)
+        return "Error: An unexpected error occurred"
 
 iface = gr.Interface(fn=generate_response, inputs="text", outputs="text", title="Generative AI Query Response")
 iface.launch()