spaces-research-think

Running

App Files Files Community

ginipick commited on Oct 17, 2024

Commit

5b8a14c

verified ·

1 Parent(s): fc11c0b

Update app.py

Browse files

Files changed (1) hide show

app.py +51 -111

app.py CHANGED Viewed

@@ -1,15 +1,16 @@
 import requests
 import concurrent.futures
-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")
@@ -88,112 +89,51 @@ def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
     with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
         return list(executor.map(format_space, spaces))
-@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">
-    <head>
-        <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>
-        <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/' + 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 = "";
-                }
-            }
-        </script>
-    </body>
-    </html>
-    """
-    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)

+import gradio as gr
+from huggingface_hub import InferenceClient
+import os
 import requests
+from typing import List, Dict, Union, Tuple
 import concurrent.futures
 # 환경 변수에서 토큰 가져오기
 HF_TOKEN = os.getenv("HF_TOKEN")
+# 추론 API 클라이언트 설정
+hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=HF_TOKEN)
 def get_headers():
     if not HF_TOKEN:
         raise ValueError("Hugging Face token not found in environment variables")
     with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
         return list(executor.map(format_space, spaces))
+def summarize_space(space: Dict) -> str:
+    system_message = "당신은 Hugging Face Space의 내용을 요약하는 AI 조수입니다. 주어진 정보를 바탕으로 간결하고 명확한 요약을 제공해주세요."
+    user_message = f"다음 Hugging Face Space를 요약해주세요: {space['name']} by {space['author']}. 좋아요 수: {space['likes']}. URL: {space['url']}"
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": user_message}
+    ]
+    response = hf_client.chat_completion(messages, max_tokens=150, temperature=0.7)
+    return response.choices[0].message.content
+def create_ui():
+    spaces_list = get_most_liked_spaces()
+    formatted_spaces = format_spaces(spaces_list)
+    with gr.Blocks() as demo:
+        gr.Markdown("# Hugging Face Most Liked Spaces")
+        with gr.Row():
+            with gr.Column(scale=1):
+                space_list = gr.List(
+                    [f"{space['name']} by {space['author']} (Likes: {space['likes']})" for space in formatted_spaces],
+                    label="Most Liked Spaces"
+                )
+                summarize_btn = gr.Button("요약")
+            with gr.Column(scale=2):
+                output = gr.Textbox(label="Space 정보", lines=10)
+                app_py_content = gr.Code(language="python", label="app.py 내용")
+        def on_select(evt: gr.SelectData):
+            selected_space = formatted_spaces[evt.index]
+            app_content = get_app_py_content(selected_space['id'])
+            return selected_space['url'], app_content
+        def on_summarize(evt: gr.SelectData):
+            selected_space = formatted_spaces[evt.index]
+            summary = summarize_space(selected_space)
+            return summary
+        space_list.select(on_select, None, [output, app_py_content])
+        summarize_btn.click(on_summarize, None, output)
+    return demo
 if __name__ == "__main__":
+    demo = create_ui()
+    demo.launch()