Update app.py

app.py

@@ -87,8 +87,7 @@ 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=5):
+async def try_claude_api(system_message, claude_messages, timeout=15):  # timeout을 15초로 증가
     try:
         start_time = time.time()
         with claude_client.messages.stream(
@@ -98,22 +97,26 @@ async def try_claude_api(system_message, claude_messages, timeout=5):
             messages=claude_messages
         ) as stream:
             collected_content = ""
-            for chunk in stream:
-                if time.time() - start_time > timeout:
+            async for chunk in stream:  # async 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
+                
+                # 각 청크마다 타임아웃 카운터 리셋
+                start_time = current_time
+                
     except Exception as e:
         print(f"Claude API error: {str(e)}")
         raise e
 
-
-        
 async def try_openai_api(openai_messages):
     try:
         stream = openai_client.chat.completions.create(
-            model="gpt-4o",
+            model="gpt-4o",  # 모델명 유지
             messages=openai_messages,
             stream=True,
             max_tokens=4096,
@@ -130,6 +133,7 @@ async def try_openai_api(openai_messages):
         print(f"OpenAI API error: {str(e)}")
         raise e
 
+
 async def generation_code(query: Optional[str], _setting: Dict[str, str], _history: Optional[History]):
     if not query or query.strip() == '':
         query = random.choice(DEMO_LIST)['description']