update to handle app not starting

app.py

@@ -1,24 +1,35 @@
 # app.py
 
 import os
+from flask import Flask, jsonify
 from utils.data_loader import download_dataset, save_metadata
 from utils.vector_utils import create_vector_db
 from config import v_auth_token, v_vector_folder, v_metadata_file
 
-if __name__ == "__main__":
+app = Flask(__name__)
+
+@app.route('/health', methods=['GET'])
+def health_check():
+    return jsonify({"status": "healthy"}), 200
+
+@app.route('/initialize', methods=['POST'])
+def initialize():
     if not v_auth_token:
-        raise ValueError("Authentication token not found. Ensure 'hkey' is set as a secret in Hugging Face Spaces.")
+        return jsonify({"error": "Authentication token not found"}), 500
 
     # Ensure writable base directory and subdirectories are initialized
     os.makedirs(v_vector_folder, exist_ok=True)
     if not os.path.exists(v_metadata_file):
-        print("Metadata file not found. Creating a new one.")
         save_metadata(v_metadata_file, {})
 
     print("Starting Vector Database Creation...")
 
     v_dataset_path = download_dataset(v_auth_token, v_metadata_file, v_vector_folder)
     if not v_dataset_path:
-        print("Vector database is up-to-date.")
+        return jsonify({"message": "Vector database is up-to-date"}), 200
     else:
         create_vector_db(v_dataset_path, v_vector_folder, v_auth_token)
+        return jsonify({"message": "Vector database created successfully"}), 200
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)

requirements.txt

@@ -3,3 +3,4 @@ huggingface-hub
 sentence-transformers
 numpy
 pandas
+flask