Update app.py

app.py

@@ -157,29 +157,42 @@ with gr.Blocks(css_paths="app.css") as demo:
                         gr.update(open=True)   # code_drawer
                     ]
 
-                    response = client.messages.create(
+                    # 스트리밍 응답 사용
+                    with client.messages.stream(
                         model="claude-3-5-sonnet-20241022",
                         max_tokens=7800,
                         system=system_message,
                         messages=claude_messages
-                    )
-                    
-                    content = response.content[0].text
-                    _history = messages_to_history([
-                        {'role': Role.SYSTEM, 'content': system_message}
-                    ] + claude_messages + [{
-                        'role': Role.ASSISTANT,
-                        'content': content
-                    }])
-                    
-                    # 최종 결과 반환
-                    yield [
-                        content,              # code_output
-                        _history,            # history
-                        send_to_sandbox(remove_code_block(content)),  # sandbox
-                        gr.update(active_key="render"),  # state_tab
-                        gr.update(open=True)  # code_drawer
-                    ]
+                    ) 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)}')