spaces-research-think

Running

App Files Files Community

ginipick commited on Oct 17, 2024

Commit

240c790

verified ·

1 Parent(s): 5efe8ef

Update app.py

Browse files

Files changed (1) hide show

app.py +31 -15

app.py CHANGED Viewed

@@ -3,9 +3,18 @@ import concurrent.futures
 from flask import Flask, render_template_string, jsonify
 from typing import List, Dict, Union
 import base64
 app = Flask(__name__)
 def get_most_liked_spaces(limit: int = 100) -> Union[List[Dict], str]:
     url = "https://huggingface.co/api/spaces"
     params = {
@@ -16,7 +25,7 @@ def get_most_liked_spaces(limit: int = 100) -> Union[List[Dict], str]:
     }
     try:
-        response = requests.get(url, params=params)
         response.raise_for_status()
         data = response.json()
@@ -32,7 +41,7 @@ def get_most_liked_spaces(limit: int = 100) -> Union[List[Dict], str]:
 def capture_thumbnail(space_id: str) -> str:
     screenshot_url = f"https://huggingface.co/spaces/{space_id}/screenshot.jpg"
     try:
-        response = requests.get(screenshot_url)
         if response.status_code == 200:
             return base64.b64encode(response.content).decode('utf-8')
     except requests.RequestException:
@@ -42,13 +51,13 @@ def capture_thumbnail(space_id: str) -> str:
 def get_app_py_content(space_id: str) -> str:
     app_py_url = f"https://huggingface.co/spaces/{space_id}/raw/main/app.py"
     try:
-        response = requests.get(app_py_url)
         if response.status_code == 200:
             return response.text
         else:
-            return "app.py file not found or inaccessible"
     except requests.RequestException:
-        return "Error fetching app.py content"
 def format_space(space: Dict) -> Dict:
     space_id = space.get('id', 'Unknown')
@@ -62,7 +71,6 @@ def format_space(space: Dict) -> Dict:
     space_url = f"https://huggingface.co/spaces/{space_id}"
     thumbnail = capture_thumbnail(space_id)
-    app_py_content = get_app_py_content(space_id)
     return {
         "id": space_id,
@@ -70,8 +78,7 @@ def format_space(space: Dict) -> Dict:
         "author": space_author,
         "likes": space_likes,
         "url": space_url,
-        "thumbnail": thumbnail,
-        "app_py_content": app_py_content
     }
 def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
@@ -83,9 +90,12 @@ def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
 @app.route('/')
 def index():
-    spaces_list = get_most_liked_spaces()
-    formatted_spaces = format_spaces(spaces_list)
     html_template = """
     <!DOCTYPE html>
     <html lang="en">
@@ -158,10 +168,13 @@ def index():
                 var selector = document.getElementById("spaceSelector");
                 var spaceId = selector.value;
                 if (spaceId) {
-                    fetch('/app_content/' + spaceId)
                         .then(response => response.json())
                         .then(data => {
                             document.getElementById("appContent").textContent = data.content;
                         });
                 } else {
                     document.getElementById("appContent").textContent = "";
@@ -174,10 +187,13 @@ def index():
     return render_template_string(html_template, spaces=formatted_spaces)
-@app.route('/app_content/<space_id>')
 def app_content(space_id):
-    content = get_app_py_content(space_id)
-    return jsonify({'content': content})
 if __name__ == "__main__":
     app.run(host='0.0.0.0', port=7860)

 from flask import Flask, render_template_string, jsonify
 from typing import List, Dict, Union
 import base64
+import os
 app = Flask(__name__)
+# 환경 변수에서 토큰 가져오기
+HF_TOKEN = os.getenv("HF_TOKEN")
+def get_headers():
+    if not HF_TOKEN:
+        raise ValueError("Hugging Face token not found in environment variables")
+    return {"Authorization": f"Bearer {HF_TOKEN}"}
 def get_most_liked_spaces(limit: int = 100) -> Union[List[Dict], str]:
     url = "https://huggingface.co/api/spaces"
     params = {
     }
     try:
+        response = requests.get(url, params=params, headers=get_headers())
         response.raise_for_status()
         data = response.json()
 def capture_thumbnail(space_id: str) -> str:
     screenshot_url = f"https://huggingface.co/spaces/{space_id}/screenshot.jpg"
     try:
+        response = requests.get(screenshot_url, headers=get_headers())
         if response.status_code == 200:
             return base64.b64encode(response.content).decode('utf-8')
     except requests.RequestException:
 def get_app_py_content(space_id: str) -> str:
     app_py_url = f"https://huggingface.co/spaces/{space_id}/raw/main/app.py"
     try:
+        response = requests.get(app_py_url, headers=get_headers())
         if response.status_code == 200:
             return response.text
         else:
+            return f"app.py file not found or inaccessible for space: {space_id}"
     except requests.RequestException:
+        return f"Error fetching app.py content for space: {space_id}"
 def format_space(space: Dict) -> Dict:
     space_id = space.get('id', 'Unknown')
     space_url = f"https://huggingface.co/spaces/{space_id}"
     thumbnail = capture_thumbnail(space_id)
     return {
         "id": space_id,
         "author": space_author,
         "likes": space_likes,
         "url": space_url,
+        "thumbnail": thumbnail
     }
 def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
 @app.route('/')
 def index():
+    try:
+        spaces_list = get_most_liked_spaces()
+        formatted_spaces = format_spaces(spaces_list)
+    except ValueError as e:
+        return str(e), 500
     html_template = """
     <!DOCTYPE html>
     <html lang="en">
                 var selector = document.getElementById("spaceSelector");
                 var spaceId = selector.value;
                 if (spaceId) {
+                    fetch('/app_content/' + encodeURIComponent(spaceId))
                         .then(response => response.json())
                         .then(data => {
                             document.getElementById("appContent").textContent = data.content;
+                        })
+                        .catch(error => {
+                            document.getElementById("appContent").textContent = "Error fetching app.py content: " + error;
                         });
                 } else {
                     document.getElementById("appContent").textContent = "";
     return render_template_string(html_template, spaces=formatted_spaces)
+@app.route('/app_content/<path:space_id>')
 def app_content(space_id):
+    try:
+        content = get_app_py_content(space_id)
+        return jsonify({'content': content})
+    except ValueError as e:
+        return jsonify({'error': str(e)}), 500
 if __name__ == "__main__":
     app.run(host='0.0.0.0', port=7860)