Update app.py

app.py

@@ -297,16 +297,18 @@ def gemini():
     if not prompt:
         return jsonify({'error': 'Prompt parameter is required'}), 400
 
-    convo = model.start_chat(history=[])
-    convo.send_message(prompt)
-    response = convo.last.text
+    try:
+        convo = model.start_chat(history=[])
+        convo.send_message(prompt)
+        response = convo.last.text
 
-    return jsonify({'response': response})
+        return jsonify({'response': response})
 
     except Exception as e:
         error_message = str(e)
         app.logger.error("Failed to generate content: %s", error_message)
         return jsonify({'error': 'Failed to generate content. Error: {}'.format(error_message)}), 500
+
     
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860, debug=True)