Update app-backup3.py

app-backup3.py

@@ -1,6 +1,8 @@
 import requests
-from flask import Flask, render_template_string
+import concurrent.futures
+from flask import Flask, render_template_string, jsonify
 from typing import List, Dict, Union
+import base64
 
 app = Flask(__name__)
 
@@ -27,29 +29,57 @@ def get_most_liked_spaces(limit: int = 100) -> Union[List[Dict], str]:
     except ValueError as e:
         return f"JSON decoding error: {str(e)}"
 
-def format_spaces(spaces: Union[List[Dict], str]) -> List[str]:
-    if isinstance(spaces, str):
-        return [spaces]  # 오류 메시지를 리스트로 반환
+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:
+        pass
+    return ""
+
+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')
+    space_name = space_id.split('/')[-1] if '/' in space_id else space_id
     
-    formatted_spaces = []
-    for space in spaces:
-        if not isinstance(space, dict):
-            formatted_spaces.append(f"Unexpected space data format: {type(space)}")
-            continue
-        
-        space_id = space.get('id', 'Unknown')
-        space_name = space_id.split('/')[-1] if '/' in space_id else space_id
-        
-        space_author = space.get('author', 'Unknown')
-        if isinstance(space_author, dict):
-            space_author = space_author.get('user', space_author.get('name', 'Unknown'))
-        
-        space_likes = space.get('likes', 'N/A')
-        
-        formatted_space = f"{space_name} by {space_author} (Likes: {space_likes})"
-        formatted_spaces.append(formatted_space)
+    space_author = space.get('author', 'Unknown')
+    if isinstance(space_author, dict):
+        space_author = space_author.get('user', space_author.get('name', 'Unknown'))
+    
+    space_likes = space.get('likes', 'N/A')
+    space_url = f"https://huggingface.co/spaces/{space_id}"
+    
+    thumbnail = capture_thumbnail(space_id)
+    app_py_content = get_app_py_content(space_id)
     
-    return formatted_spaces
+    return {
+        "id": space_id,
+        "name": space_name,
+        "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]:
+    if isinstance(spaces, str):
+        return [{"error": spaces}]
+    
+    with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
+        return list(executor.map(format_space, spaces))
 
 @app.route('/')
 def index():
@@ -63,19 +93,91 @@ def index():
         <meta charset="UTF-8">
         <meta name="viewport" content="width=device-width, initial-scale=1.0">
         <title>Hugging Face Most Liked Spaces</title>
+        <style>
+            body { font-family: Arial, sans-serif; margin: 0; padding: 0; }
+            .container { max-width: 1200px; margin: 0 auto; padding: 20px; }
+            .space-item { margin-bottom: 20px; }
+            .space-item img { max-width: 200px; max-height: 200px; }
+            .tab { overflow: hidden; border: 1px solid #ccc; background-color: #f1f1f1; }
+            .tab button { background-color: inherit; float: left; border: none; outline: none; cursor: pointer; padding: 14px 16px; transition: 0.3s; }
+            .tab button:hover { background-color: #ddd; }
+            .tab button.active { background-color: #ccc; }
+            .tabcontent { display: none; padding: 6px 12px; border: 1px solid #ccc; border-top: none; }
+            pre { white-space: pre-wrap; word-wrap: break-word; }
+        </style>
     </head>
     <body>
-        <h1>Hugging Face Most Liked Spaces</h1>
-        <ol>
-        {% for space in spaces %}
-            <li>{{ space }}</li>
-        {% endfor %}
-        </ol>
+        <div class="container">
+            <h1>Hugging Face Most Liked Spaces</h1>
+            <div class="tab">
+                <button class="tablinks active" onclick="openTab(event, 'MostLiked')">Most Liked</button>
+                <button class="tablinks" onclick="openTab(event, 'AppContent')">App.py Content</button>
+            </div>
+            <div id="MostLiked" class="tabcontent" style="display: block;">
+                <ol>
+                {% for space in spaces %}
+                    <li class="space-item">
+                        <a href="{{ space.url }}" target="_blank">
+                            {{ space.name }} by {{ space.author }} (Likes: {{ space.likes }})
+                        </a>
+                        {% if space.thumbnail %}
+                            <br>
+                            <img src="data:image/jpeg;base64,{{ space.thumbnail }}" alt="{{ space.name }} thumbnail">
+                        {% endif %}
+                    </li>
+                {% endfor %}
+                </ol>
+            </div>
+            <div id="AppContent" class="tabcontent">
+                <h2>Select a space to view its app.py content:</h2>
+                <select id="spaceSelector" onchange="showAppContent()">
+                    <option value="">Select a space</option>
+                    {% for space in spaces %}
+                        <option value="{{ space.id }}">{{ space.name }} by {{ space.author }}</option>
+                    {% endfor %}
+                </select>
+                <pre id="appContent"></pre>
+            </div>
+        </div>
+        <script>
+            function openTab(evt, tabName) {
+                var i, tabcontent, tablinks;
+                tabcontent = document.getElementsByClassName("tabcontent");
+                for (i = 0; i < tabcontent.length; i++) {
+                    tabcontent[i].style.display = "none";
+                }
+                tablinks = document.getElementsByClassName("tablinks");
+                for (i = 0; i < tablinks.length; i++) {
+                    tablinks[i].className = tablinks[i].className.replace(" active", "");
+                }
+                document.getElementById(tabName).style.display = "block";
+                evt.currentTarget.className += " active";
+            }
+
+            function showAppContent() {
+                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 = "";
+                }
+            }
+        </script>
     </body>
     </html>
     """
     
     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)