Update app.py

app.py

@@ -172,92 +172,202 @@ async def try_openai_api(openai_messages):
         print(f"OpenAI API error: {str(e)}")
         raise e
 
+
+def analyze_code(code: str) -> str:
+    """코드 분석 결과를 HTML 형식으로 반환"""
+    analysis = []
+    
+    # 1. 사용된 라이브러리 분석
+    imports = []
+    for line in code.split('\n'):
+        if line.startswith('import ') or line.startswith('from '):
+            imports.append(line.strip())
+    
+    if imports:
+        analysis.append("<h2>📚 사용된 라이브러리</h2>")
+        analysis.append("<ul>")
+        for imp in imports:
+            analysis.append(f"<li><code>{imp}</code></li>")
+        analysis.append("</ul>")
+    
+    # 2. 함수 분석
+    functions = []
+    current_func = []
+    in_function = False
+    
+    for line in code.split('\n'):
+        if line.strip().startswith('def '):
+            if current_func:
+                functions.append('\n'.join(current_func))
+                current_func = []
+            in_function = True
+        if in_function:
+            current_func.append(line)
+        if in_function and not line.strip():
+            in_function = False
+            if current_func:
+                functions.append('\n'.join(current_func))
+                current_func = []
+    
+    if functions:
+        analysis.append("<h2>🔧 주요 함수</h2>")
+        for func in functions:
+            func_name = func.split('def ')[1].split('(')[0]
+            analysis.append(f"<h3><code>{func_name}</code></h3>")
+            analysis.append(f"<p>{get_function_description(func)}</p>")
+    
+    # 3. UI 컴포넌트 분석
+    ui_components = []
+    for line in code.split('\n'):
+        if 'gr.' in line:
+            component = line.split('gr.')[1].split('(')[0]
+            if component not in ui_components:
+                ui_components.append(component)
+    
+    if ui_components:
+        analysis.append("<h2>🎨 UI 컴포넌트</h2>")
+        analysis.append("<ul>")
+        for component in ui_components:
+            analysis.append(f"<li><strong>{component}</strong>: {get_component_description(component)}</li>")
+        analysis.append("</ul>")
+    
+    # 4. 실행 방법
+    analysis.append("<h2>▶️ 실행 방법</h2>")
+    analysis.append("<ol>")
+    analysis.append("<li>'실행하기' 버튼을 클릭하여 Hugging Face Space에 배포</li>")
+    analysis.append("<li>생성된 링크를 클릭하여 애플리케이션 실행</li>")
+    analysis.append("</ol>")
+    
+    return "\n".join(analysis)
+
+def get_function_description(func: str) -> str:
+    """함수의 목적을 설명하는 문자열 반환"""
+    if 'get_multiplication_table' in func:
+        return "입력받은 숫자의 구구단을 계산하여 문자열로 반환"
+    elif 'get_all_tables' in func:
+        return "2단부터 9단까지 전체 구구단을 생성하여 반환"
+    # 다른 함수들에 대한 설명 추가
+    return "함수의 기능 설명"
+
+def get_component_description(component: str) -> str:
+    """UI 컴포넌트에 대한 설명 반환"""
+    descriptions = {
+        'Number': '숫자 입력 필드',
+        'Button': '클릭 가능한 버튼',
+        'Textbox': '텍스트 출력 영역',
+        'Markdown': '마크다운 형식의 텍스트 표시',
+        'Row': '수평 방향 레이아웃',
+        'Column': '수직 방향 레이아웃',
+        'Blocks': '전체 UI 컨테이너',
+        'Image': '이미지 표시 컴포넌트',
+        'File': '파일 업로드 컴포넌트',
+        'Slider': '슬라이더 입력 컴포넌트',
+        'Dropdown': '드롭다운 선택 컴포넌트',
+        'Radio': '라디오 버튼 그룹',
+        'Checkbox': '체크박스 컴포넌트',
+        'Audio': '오디오 재생/녹음 컴포넌트',
+        'Video': '비디오 재생 컴포넌트',
+        'HTML': 'HTML 콘텐츠 표시',
+        'JSON': 'JSON 데이터 표시',
+        'DataFrame': '데이터프레임 표시',
+        'Plot': '그래프/차트 표시',
+        'Label': '레이블 텍스트 표시'
+    }
+    return descriptions.get(component, '컴포넌트 설명')
+
+
+
 class Demo:
     def __init__(self):
         pass
 
-    async def generation_code(self, query: Optional[str], _setting: Dict[str, str], _history: Optional[History]):
-        if not query or query.strip() == '':
-            query = random.choice(DEMO_LIST)['description']
-        
-        if _history is None:
-            _history = []
-        
-        messages = history_to_messages(_history, _setting['system'])
-        system_message = messages[0]['content']
-        
-        claude_messages = [
-            {"role": msg["role"] if msg["role"] != "system" else "user", "content": msg["content"]} 
-            for msg in messages[1:] + [{'role': Role.USER, 'content': query}]
-            if msg["content"].strip() != ''
-        ]
-        
-        openai_messages = [{"role": "system", "content": system_message}]
-        for msg in messages[1:]:
-            openai_messages.append({
-                "role": msg["role"],
-                "content": msg["content"]
-            })
-        openai_messages.append({"role": "user", "content": query})
-        
-        try:
-            yield [
-                "Generating code...",
-                _history,
-                None,
-                gr.update(active_key="loading"),
-                gr.update(open=True)
-            ]
-            await asyncio.sleep(0)
-
-            collected_content = None
-            try:
-                async for content in try_claude_api(system_message, claude_messages):
-                    yield [
-                        content,
-                        _history,
-                        None,
-                        gr.update(active_key="loading"),
-                        gr.update(open=True)
-                    ]
-                    await asyncio.sleep(0)
-                collected_content = content
-                
-            except Exception as claude_error:
-                print(f"Falling back to OpenAI API due to Claude error: {str(claude_error)}")
-                
-                async for content in try_openai_api(openai_messages):
-                    yield [
-                        content,
-                        _history,
-                        None,
-                        gr.update(active_key="loading"),
-                        gr.update(open=True)
-                    ]
-                    await asyncio.sleep(0)
-                collected_content = content
-
-            if collected_content:
-                _history = messages_to_history([
-                    {'role': Role.SYSTEM, 'content': system_message}
-                ] + claude_messages + [{
-                    'role': Role.ASSISTANT,
-                    'content': collected_content
-                }])
-                
-                yield [
-                    collected_content,
+# Demo 클래스의 generation_code 메서드 수정
+     async def generation_code(self, query: Optional[str], _setting: Dict[str, str], _history: Optional[History]):
+          if not query or query.strip() == '':
+               query = random.choice(DEMO_LIST)['description']
+          
+          if _history is None:
+               _history = []
+          
+          messages = history_to_messages(_history, _setting['system'])
+          system_message = messages[0]['content']
+          
+          claude_messages = [
+               {"role": msg["role"] if msg["role"] != "system" else "user", "content": msg["content"]} 
+               for msg in messages[1:] + [{'role': Role.USER, 'content': query}]
+               if msg["content"].strip() != ''
+          ]
+          
+          openai_messages = [{"role": "system", "content": system_message}]
+          for msg in messages[1:]:
+               openai_messages.append({
+                    "role": msg["role"],
+                    "content": msg["content"]
+               })
+          openai_messages.append({"role": "user", "content": query})
+          
+          try:
+               yield [
+                    "Generating code...",
                     _history,
-                    send_to_sandbox(remove_code_block(collected_content)),
-                    gr.update(active_key="render"),
+                    None,
+                    gr.update(active_key="loading"),
                     gr.update(open=True)
-                ]
-            else:
-                raise ValueError("No content was generated from either API")
-                
-        except Exception as e:
-            print(f"Error details: {str(e)}")
-            raise ValueError(f'Error calling APIs: {str(e)}')
+               ]
+               await asyncio.sleep(0)
+
+               collected_content = None
+               try:
+                    async for content in try_claude_api(system_message, claude_messages):
+                         code = content
+                         analysis = analyze_code(code)
+                         yield [
+                              code,
+                              _history,
+                              analysis,  # 분석 결과를 HTML로 전달
+                              gr.update(active_key="loading"),
+                              gr.update(open=True)
+                         ]
+                         await asyncio.sleep(0)
+                    collected_content = code
+                    
+               except Exception as claude_error:
+                    print(f"Falling back to OpenAI API due to Claude error: {str(claude_error)}")
+                    
+                    async for content in try_openai_api(openai_messages):
+                         code = content
+                         analysis = analyze_code(code)
+                         yield [
+                              code,
+                              _history,
+                              analysis,  # 분석 결과를 HTML로 전달
+                              gr.update(active_key="loading"),
+                              gr.update(open=True)
+                         ]
+                         await asyncio.sleep(0)
+                    collected_content = code
+
+               if collected_content:
+                    _history = messages_to_history([
+                         {'role': Role.SYSTEM, 'content': system_message}
+                    ] + claude_messages + [{
+                         'role': Role.ASSISTANT,
+                         'content': collected_content
+                    }])
+                    
+                    yield [
+                         collected_content,
+                         _history,
+                         analyze_code(collected_content),  # 최종 분석 결과를 HTML로 전달
+                         gr.update(active_key="render"),
+                         gr.update(open=True)
+                    ]
+               else:
+                    raise ValueError("No content was generated from either API")
+                    
+          except Exception as e:
+               print(f"Error details: {str(e)}")
+               raise ValueError(f'Error calling APIs: {str(e)}')
 
     def clear_history(self):
         return []