Update app.py

app.py

@@ -11,6 +11,7 @@ from gradio_client import Client
 import time
 import threading
 import json
+import re
 
 HF_TOKEN = os.getenv("HF_TOKEN")
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=HF_TOKEN)
@@ -61,77 +62,15 @@ def get_space_structure(space_id: str) -> Dict:
         return {"error": f"API request error: {str(e)}"}
 
 def format_tree_structure(tree_data: Dict, indent: str = "") -> str:
-    formatted = f"{indent}{'📁' if tree_data['type'] == 'directory' else '📄'} {tree_data['name']}\n"
-    if tree_data["type"] == "directory":
-        for child in sorted(tree_data.get("children", []), key=lambda x: (x["type"] != "directory", x["name"])):
+    if "error" in tree_data:
+        return tree_data["error"]
+    
+    formatted = f"{indent}{'📁' if tree_data.get('type') == 'directory' else '📄'} {tree_data.get('name', 'Unknown')}\n"
+    if tree_data.get("type") == "directory":
+        for child in sorted(tree_data.get("children", []), key=lambda x: (x.get("type", "") != "directory", x.get("name", ""))):
             formatted += format_tree_structure(child, indent + "  ")
     return formatted
 
-
-
-
-
-
-
-def analyze_space(url: str, progress=gr.Progress()):
-    try:
-        space_id = url.split('spaces/')[-1]
-        
-        progress(0.1, desc="파일 구조 분석 중...")
-        tree_structure = get_space_structure(space_id)
-        tree_view = format_tree_structure(tree_structure)
-        
-        progress(0.3, desc="app.py 내용 가져오는 중...")
-        app_content = get_file_content(space_id, "app.py")
-        
-        progress(0.4, desc="코드 요약 중...")
-        summary = summarize_code(app_content)
-        
-        progress(0.6, desc="코드 분석 중...")
-        analysis = analyze_code(app_content)
-        
-        progress(0.8, desc="사용법 설명 생성 중...")
-        usage = explain_usage(app_content)
-        
-        progress(1.0, desc="완료")
-        return summary, analysis, usage, app_content, tree_view, tree_structure, space_id
-    except Exception as e:
-        print(f"Error in analyze_space: {str(e)}")
-        print(traceback.format_exc())
-        return f"오류가 발생했습니다: {str(e)}", "", "", "", "", None, ""
-
-def respond(
-    message: str,
-    history: List[Tuple[str, str]],
-    system_message: str = "",
-    max_tokens: int = 4000,
-    temperature: float = 0.7,
-    top_p: float = 0.9,
-):
-    system_prefix = """반드시 한글로 답변할것. 너는 주어진 소스코드를 기반으로 "서비스 사용 설명 및 안내, qna를 하는 역할이다". 아주 친절하고 자세하게 4000토큰 이상 작성하라. 너는 코드를 기반으로 사용 설명 및 질의 응답을 진행하며, 이용자에게 도움을 주어야 한다. 이용자가 궁금해 할 만 한 내용에 친절하게 알려주도록 하라. 코드 전체 내용에 대해서는 보안을 유지하고, 키 값 및 엔드포인트와 구체적인 모델은 공개하지 마라."""
-
-    messages = [{"role": "system", "content": f"{system_prefix} {system_message}"}]
-    for user, assistant in history:
-        messages.append({"role": "user", "content": user})
-        messages.append({"role": "assistant", "content": assistant})
-    messages.append({"role": "user", "content": message})
-
-    response = ""
-    for message in hf_client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.get('content', None)
-        if token:
-            response += token.strip("")
-            yield response
-
-
-
-
 def summarize_code(app_content: str):
     system_message = "당신은 Python 코드를 분석하고 요약하는 AI 조수입니다. 주어진 코드를 3줄 이내로 간결하게 요약해주세요."
     user_message = f"다음 Python 코드를 3줄 이내로 요약해주세요:\n\n{app_content}"
@@ -187,8 +126,14 @@ def analyze_space(url: str, progress=gr.Progress()):
     try:
         space_id = url.split('spaces/')[-1]
         
+        # Space ID 유효성 검사
+        if not re.match(r'^[\w.-]{1,96}$', space_id) or space_id.startswith(('-', '.')) or space_id.endswith(('-', '.')):
+            raise ValueError(f"Invalid Space ID: {space_id}")
+        
         progress(0.1, desc="파일 구조 분석 중...")
         tree_structure = get_space_structure(space_id)
+        if "error" in tree_structure:
+            raise ValueError(tree_structure["error"])
         tree_view = format_tree_structure(tree_structure)
         
         progress(0.3, desc="app.py 내용 가져오는 중...")
@@ -201,8 +146,6 @@ def analyze_space(url: str, progress=gr.Progress()):
         print(traceback.format_exc())
         return f"오류가 발생했습니다: {str(e)}", "", None, ""
 
-
-
 def create_ui():
     try:
         css = """
@@ -244,7 +187,7 @@ def create_ui():
         """
 
         with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
-            gr.Markdown("# mouse lightning: HuggingFace")
+            gr.Markdown("# HuggingFace Space Analyzer")
             
             with gr.Tabs(elem_classes="main-tabs") as tabs:
                 with gr.TabItem("분석"):
@@ -289,7 +232,9 @@ def create_ui():
                         top_p = gr.Slider(minimum=0, maximum=1, value=0.9, label="Top P")
                     
                     examples = [
-
+                        ["패션 코드 실행"],
+                        ["UHD 이미지 코드 실행"],
+                        ["MixGEN 코드 실행"],
                         ["상세한 사용 방법을 마치 화면을 보면서 설명하듯이 4000 토큰 이상 자세히 설명하라"],
                         ["FAQ 20건을 상세하게 작성하라. 4000토큰 이상 사용하라."],
                         ["사용 방법과 차별점, 특징, 강점을 중심으로 4000 토큰 이상 유튜브 영상 스크립트 형태로 작성하라"],
@@ -407,4 +352,4 @@ if __name__ == "__main__":
         )
     except Exception as e:
         print(f"Error in main: {str(e)}")
-        print(traceback.format_exc())
+        print(traceback.format_exc())