Update app.py

app.py

@@ -140,9 +140,8 @@ with gr.Blocks(css_paths="app.css") as demo:
                     _history = []
                 
                 messages = history_to_messages(_history, _setting['system'])
-                system_message = messages[0]['content']  # Get system message
+                system_message = messages[0]['content']
                 
-                # Convert remaining messages to Claude format (excluding system message)
                 claude_messages = [
                     {"role": msg["role"] if msg["role"] != "system" else "user", "content": msg["content"]} 
                     for msg in messages[1:] + [{'role': Role.USER, 'content': query}]
@@ -151,15 +150,15 @@ with gr.Blocks(css_paths="app.css") as demo:
                 try:
                     # 중간 상태 표시
                     yield {
-                        code_output: "Generating code...",
-                        state_tab: gr.update(active_key="loading"),
-                        code_drawer: gr.update(open=True),
+                        "code_output": "Generating code...",
+                        "state_tab": gr.update(active_key="loading"),
+                        "code_drawer": gr.update(open=True),
                     }
 
                     response = client.messages.create(
                         model="claude-3-sonnet-20240229",
                         max_tokens=4096,
-                        system=system_message,  # System message as separate parameter
+                        system=system_message,
                         messages=claude_messages
                     )
                     
@@ -173,19 +172,27 @@ with gr.Blocks(css_paths="app.css") as demo:
                     
                     # 최종 결과 반환
                     yield {
-                        code_output: content,
-                        history: _history,
-                        sandbox: send_to_sandbox(remove_code_block(content)),
-                        state_tab: gr.update(active_key="render"),
-                        code_drawer: gr.update(open=False),
+                        "code_output": content,
+                        "history": _history,
+                        "sandbox": send_to_sandbox(remove_code_block(content)),
+                        "state_tab": gr.update(active_key="render"),
+                        "code_drawer": gr.update(open=True),
                     }
                     
                 except Exception as e:
                     raise ValueError(f'Error calling Claude API: {str(e)}')
 
-            btn.click(generation_code,
-                      inputs=[input, setting, history],
-                      outputs=[code_output, history, sandbox, state_tab, code_drawer])
+            btn.click(
+                generation_code,
+                inputs=[input, setting, history],
+                outputs={
+                    "code_output": code_output,
+                    "history": history,
+                    "sandbox": sandbox,
+                    "state_tab": state_tab,
+                    "code_drawer": code_drawer
+                }
+            )
             
             clear_btn.click(clear_history, inputs=[], outputs=[history])