Update app-backup1.py

app-backup1.py

@@ -175,89 +175,157 @@ async def try_openai_api(openai_messages):
 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,
+    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,
+                              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,
+                              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),
+                         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 analyze_code(code: str) -> str:
+          """코드 분석 결과를 Markdown 형식으로 반환"""
+          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("## 📚 사용된 라이브러리")
+               for imp in imports:
+                    analysis.append(f"- `{imp}`")
+               analysis.append("")
+          
+          # 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("## 🔧 주요 함수")
+               for func in functions:
+                    func_name = func.split('def ')[1].split('(')[0]
+                    analysis.append(f"### `{func_name}`")
+                    analysis.append(get_function_description(func))
+                    analysis.append("")
+          
+          # 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("## 🎨 UI 컴포넌트")
+               for component in ui_components:
+                    analysis.append(f"- **{component}**: {get_component_description(component)}")
+               analysis.append("")
+          
+          # 4. 실행 방법
+          analysis.append("## ▶️ 실행 방법")
+          analysis.append("1. '실행하기' 버튼을 클릭하여 Hugging Face Space에 배포")
+          analysis.append("2. 생성된 링크를 클릭하여 애플리케이션 실행")
+          
+          return "\n".join(analysis)
 
     def clear_history(self):
         return []