Update app.py

app.py

@@ -86,8 +86,7 @@ YOUR_OPENAI_TOKEN = os.getenv('OPENAI_API_KEY')
 claude_client = anthropic.Anthropic(api_key=YOUR_ANTHROPIC_TOKEN)
 openai_client = openai.OpenAI(api_key=YOUR_OPENAI_TOKEN)
 
-
-async def try_claude_api(system_message, claude_messages, timeout=15):  # timeout을 15초로 증가
+async def try_claude_api(system_message, claude_messages, timeout=15):
     try:
         start_time = time.time()
         with claude_client.messages.stream(
@@ -97,14 +96,15 @@ async def try_claude_api(system_message, claude_messages, timeout=15):  # timeou
             messages=claude_messages
         ) as stream:
             collected_content = ""
-            async for chunk in stream:  # async for 사용
+            for chunk in stream:  # async for 제거, 일반 for 사용
                 current_time = time.time()
                 if current_time - start_time > timeout:
                     print(f"Claude API response time: {current_time - start_time:.2f} seconds")
                     raise TimeoutError("Claude API timeout")
                 if chunk.type == "content_block_delta":
                     collected_content += chunk.delta.text
-                    yield collected_content
+                    yield collected_content  # 각 청크마다 즉시 yield
+                    await asyncio.sleep(0)  # 스트리밍을 위한 비동기 양보
                 
                 # 각 청크마다 타임아웃 카운터 리셋
                 start_time = current_time
@@ -112,6 +112,8 @@ async def try_claude_api(system_message, claude_messages, timeout=15):  # timeou
     except Exception as e:
         print(f"Claude API error: {str(e)}")
         raise e
+        
+
 
 async def try_openai_api(openai_messages):
     try:
@@ -161,6 +163,7 @@ async def generation_code(query: Optional[str], _setting: Dict[str, str], _histo
     openai_messages.append({"role": "user", "content": query})
     
     try:
+        # 먼저 코드 뷰어를 열기
         yield [
             "Generating code...",
             _history,
@@ -168,6 +171,7 @@ async def generation_code(query: Optional[str], _setting: Dict[str, str], _histo
             gr.update(active_key="loading"),
             gr.update(open=True)
         ]
+        await asyncio.sleep(0)  # UI 업데이트를 위한 양보
 
         collected_content = None
         # Claude API 시도
@@ -180,7 +184,8 @@ async def generation_code(query: Optional[str], _setting: Dict[str, str], _histo
                     gr.update(active_key="loading"),
                     gr.update(open=True)
                 ]
-                collected_content = content
+                await asyncio.sleep(0)  # UI 업데이트를 위한 양보
+            collected_content = content
             
         except Exception as claude_error:
             print(f"Falling back to OpenAI API due to Claude error: {str(claude_error)}")
@@ -194,7 +199,8 @@ async def generation_code(query: Optional[str], _setting: Dict[str, str], _histo
                     gr.update(active_key="loading"),
                     gr.update(open=True)
                 ]
-                collected_content = content
+                await asyncio.sleep(0)  # UI 업데이트를 위한 양보
+            collected_content = content
 
         if collected_content:
             _history = messages_to_history([