Update main.py

main.py

@@ -115,8 +115,25 @@ def video_feed():
 def home():
     return jsonify(message='Welcome to the Hugging Face Space!')
 
+from flask import Flask, jsonify
+import os
+
+app = Flask(__name__)
+
 @app.route('/get_frames', methods=['GET'])
 def get_frames():
+    instance_folder = app.instance_path
+    db_file_path = os.path.join(instance_folder, 'faces.db')
+
+    # Check if the instance folder is present
+    if not os.path.exists(instance_folder):
+        return jsonify({'error': 'Instance folder not present'}), 500
+
+    # Check if the faces.db file exists
+    if not os.path.exists(db_file_path):
+        return jsonify({'error': 'faces.db not present in the instance folder'}), 500
+
+    # Continue with the normal operation if everything is present
     frames = Frame.query.all()
     frames_data = []
 
@@ -128,6 +145,14 @@ def get_frames():
     
     return jsonify(frames_data)
 
+# Assuming you have a Frame model defined somewhere in your code
+class Frame:
+    pass  # Your Frame model definition goes here
+
+if __name__ == '__main__':
+    app.run(debug=True)
+
+
 if __name__ == '__main__':
     with app.app_context():
         create_database()