Update app.py

app.py

@@ -175,12 +175,13 @@ class Demo:
         pass
 
     async def generation_code(self, query: Optional[str], _setting: Dict[str, str], _history: Optional[History]):
-        if not query or query.strip() == '':
-            query = random.choice(DEMO_LIST)['description']
-        
-        if _history is None:
-            _history = []
-        
+    if not query or query.strip() == '':
+        query = random.choice(DEMO_LIST)['description']
+    
+    if _history is None:
+        _history = []
+    
+    try:
         messages = history_to_messages(_history, _setting['system'])
         system_message = messages[0]['content']
         
@@ -198,33 +199,36 @@ class Demo:
             })
         openai_messages.append({"role": "user", "content": query})
         
+        yield [
+            "Generating code...",
+            _history,
+            None,
+            gr.update(active_key="loading"),
+            gr.update(open=True)
+        ]
+        
+        collected_content = None
+        error_message = None
+        
+        # Claude API 시도
         try:
-            yield [
-                "Generating code...",
-                _history,
-                None,
-                gr.update(active_key="loading"),
-                gr.update(open=True)
-            ]
-            await asyncio.sleep(0)
-
-            collected_content = None
-            try:
-                async for content in try_claude_api(system_message, claude_messages):
-                    yield [
-                        content,
-                        _history,
-                        None,
-                        gr.update(active_key="loading"),
-                        gr.update(open=True)
-                    ]
-                    await asyncio.sleep(0)
+            async for content in try_claude_api(system_message, claude_messages):
                 collected_content = content
-                
-            except Exception as claude_error:
-                print(f"Falling back to OpenAI API due to Claude error: {str(claude_error)}")
-                
+                yield [
+                    content,
+                    _history,
+                    None,
+                    gr.update(active_key="loading"),
+                    gr.update(open=True)
+                ]
+        except Exception as claude_error:
+            print(f"Claude API error: {str(claude_error)}")
+            error_message = str(claude_error)
+            
+            # OpenAI API 시도
+            try:
                 async for content in try_openai_api(openai_messages):
+                    collected_content = content
                     yield [
                         content,
                         _history,
@@ -232,30 +236,37 @@ class Demo:
                         gr.update(active_key="loading"),
                         gr.update(open=True)
                     ]
-                    await asyncio.sleep(0)
-                collected_content = content
-
-            if collected_content:
-                _history = messages_to_history([
-                    {'role': Role.SYSTEM, 'content': system_message}
-                ] + claude_messages + [{
-                    'role': Role.ASSISTANT,
-                    'content': collected_content
-                }])
-                
-                yield [
-                    collected_content,
-                    _history,
-                    send_to_sandbox(remove_code_block(collected_content)),
-                    gr.update(active_key="render"),
-                    gr.update(open=True)
-                ]
-            else:
-                raise ValueError("No content was generated from either API")
-                
-        except Exception as e:
-            print(f"Error details: {str(e)}")
-            raise ValueError(f'Error calling APIs: {str(e)}')
+            except Exception as openai_error:
+                print(f"OpenAI API error: {str(openai_error)}")
+                error_message = f"Both APIs failed: Claude - {str(claude_error)}, OpenAI - {str(openai_error)}"
+        
+        if collected_content:
+            _history = messages_to_history([
+                {'role': Role.SYSTEM, 'content': system_message}
+            ] + claude_messages + [{
+                'role': Role.ASSISTANT,
+                'content': collected_content
+            }])
+            
+            yield [
+                collected_content,
+                _history,
+                send_to_sandbox(remove_code_block(collected_content)),
+                gr.update(active_key="render"),
+                gr.update(open=True)
+            ]
+        else:
+            raise ValueError(error_message or "No content was generated")
+            
+    except Exception as e:
+        print(f"Error details: {str(e)}")
+        yield [
+            f"Error: {str(e)}",
+            _history,
+            None,
+            gr.update(active_key="empty"),
+            gr.update(open=False)
+        ]
 
     def clear_history(self):
         return []