Update app.py

app.py

@@ -79,6 +79,70 @@ with gr.Blocks(css_paths="app.css") as demo:
         "system": SystemPrompt,
     })
 
+    def generation_code(query: Optional[str], _setting: Dict[str, str], _history: Optional[History]):
+        if query is None:
+            query = ''
+        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}]
+        ]
+        
+        try:
+            # 중간 상태 표시
+            yield [
+                "Generating code...",  # code_output
+                _history,              # history
+                None,                  # sandbox
+                gr.update(active_key="loading"),  # state_tab
+                gr.update(open=True)   # code_drawer
+            ]
+
+            # 스트리밍 응답 사용
+            with client.messages.stream(
+                model="claude-3-5-sonnet-20241022",
+                max_tokens=7800,
+                system=system_message,
+                messages=claude_messages
+            ) as stream:
+                collected_content = ""
+                for chunk in stream:
+                    if chunk.type == "content_block_delta":
+                        delta = chunk.delta.text
+                        collected_content += delta
+                        
+                        yield [
+                            collected_content,  # code_output
+                            _history,          # history
+                            None,              # sandbox (아직 완성되지 않았으므로 None)
+                            gr.update(active_key="loading"),  # state_tab
+                            gr.update(open=True)  # code_drawer
+                        ]
+                
+                # 최종 결과 반환
+                _history = messages_to_history([
+                    {'role': Role.SYSTEM, 'content': system_message}
+                ] + claude_messages + [{
+                    'role': Role.ASSISTANT,
+                    'content': collected_content
+                }])
+                
+                yield [
+                    collected_content,  # code_output
+                    _history,          # history
+                    send_to_sandbox(remove_code_block(collected_content)),  # sandbox
+                    gr.update(active_key="render"),  # state_tab
+                    gr.update(open=True)  # code_drawer
+                ]
+            
+        except Exception as e:
+            raise ValueError(f'Error calling Claude API: {str(e)}')
+
     with ms.Application() as app:
         with antd.ConfigProvider():
             with antd.Row(gutter=[32, 12]) as layout:
@@ -160,70 +224,6 @@ with gr.Blocks(css_paths="app.css") as demo:
             
             clear_btn.click(clear_history, inputs=[], outputs=[history])
 
-            def generation_code(query: Optional[str], _setting: Dict[str, str], _history: Optional[History]):
-                if query is None:
-                    query = ''
-                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}]
-                ]
-                
-                try:
-                    # 중간 상태 표시
-                    yield [
-                        "Generating code...",  # code_output
-                        _history,              # history
-                        None,                  # sandbox
-                        gr.update(active_key="loading"),  # state_tab
-                        gr.update(open=True)   # code_drawer
-                    ]
-
-                    # 스트리밍 응답 사용
-                    with client.messages.stream(
-                        model="claude-3-5-sonnet-20241022",
-                        max_tokens=7800,
-                        system=system_message,
-                        messages=claude_messages
-                    ) as stream:
-                        collected_content = ""
-                        for chunk in stream:
-                            if chunk.type == "content_block_delta":
-                                delta = chunk.delta.text
-                                collected_content += delta
-                                
-                                yield [
-                                    collected_content,  # code_output
-                                    _history,          # history
-                                    None,              # sandbox (아직 완성되지 않았으므로 None)
-                                    gr.update(active_key="loading"),  # state_tab
-                                    gr.update(open=True)  # code_drawer
-                                ]
-                        
-                        # 최종 결과 반환
-                        _history = messages_to_history([
-                            {'role': Role.SYSTEM, 'content': system_message}
-                        ] + claude_messages + [{
-                            'role': Role.ASSISTANT,
-                            'content': collected_content
-                        }])
-                        
-                        yield [
-                            collected_content,  # code_output
-                            _history,          # history
-                            send_to_sandbox(remove_code_block(collected_content)),  # sandbox
-                            gr.update(active_key="render"),  # state_tab
-                            gr.update(open=True)  # code_drawer
-                        ]
-                    
-                except Exception as e:
-                    raise ValueError(f'Error calling Claude API: {str(e)}')
-
 
 if __name__ == "__main__":
     demo.queue(default_concurrency_limit=20).launch(ssr_mode=False)