Update main.py

main.py

@@ -8,6 +8,8 @@ def hello():
 
 from flask import Flask, jsonify, request
 from datasets import load_dataset, Dataset
+from huggingface_hub import hf_hub_download
+import requests
 import pandas as pd
 import os
 
@@ -25,13 +27,31 @@ def get_leaderboard():
         df = pd.DataFrame(columns=HEADERS)
         return jsonify(df.to_dict(orient="records"))
 
-from flask import send_from_directory
 
-@app.route("/file/<filename>", methods=["GET"])
+
+
+@app.route("/file/<path:filename>", methods=["GET"])
 def get_file(filename):
-    data_dir = os.path.join(os.getcwd(), "")
+    if request.headers.get("Authorization") != f"Bearer {HF_TOKEN}":
+        return jsonify({"error": "Unauthorized"}), 403
     try:
-        return send_from_directory(data_dir, filename, as_attachment=False)
-    except FileNotFoundError:
-        return jsonify({"error": f"File '{filename}' not found"}), 404
+        # Download from HF dataset repo (metadata)
+        local_path = hf_hub_download(
+            repo_id="saraht14/metadata",
+            filename=filename,
+            repo_type="dataset",
+            token=HF_TOKEN
+        )
+
+        with open(local_path, "r") as f:
+            content = f.read()
+
+        return jsonify({
+            "filename": filename,
+            "content": content
+        })
+
+    except Exception as e:
+        return jsonify({"error": f"Failed to fetch file: {str(e)}"}), 500
+