Update app.py

app.py

@@ -256,35 +256,28 @@ class Demo:
     def __init__(self):
         pass
 
-    async def generation_code(self, query: Optional[str], _setting: Dict[str, str], _history: Optional[History]):
+    async def generation_code(self, query: Optional[str], _setting: Dict[str, str]):
         if not query or query.strip() == '':
             query = get_random_placeholder()
-        
-        if _history is None:
-            _history = []
-        
-        messages = history_to_messages(_history, _setting['system'])
+    
+        messages = [{'role': Role.SYSTEM, 'content': _setting['system']}]
+        messages.append({'role': Role.USER, 'content': query})
+    
         system_message = messages[0]['content']
-        
+    
         claude_messages = [
-            {"role": msg["role"] if msg["role"] != "system" else "user", "content": msg["content"]} 
-            for msg in messages[1:] + [{'role': Role.USER, 'content': query}]
-            if msg["content"].strip() != ''
+            {"role": "user", "content": query}
+        ]
+    
+        openai_messages = [
+            {"role": "system", "content": system_message},
+            {"role": "user", "content": query}
         ]
-        
-        openai_messages = [{"role": "system", "content": system_message}]
-        for msg in messages[1:]:
-            openai_messages.append({
-                "role": msg["role"],
-                "content": msg["content"]
-            })
-        openai_messages.append({"role": "user", "content": query})
         
         try:
             # 로딩 시작
             yield [
                 "",  # 빈 문자열로 변경 (코드 출력하지 않음)
-                _history,
                 None,
                 gr.update(active_key="loading"),
                 gr.update(open=True)
@@ -296,7 +289,6 @@ class Demo:
                 async for content in try_claude_api(system_message, claude_messages):
                     yield [
                         "",  # 빈 문자열로 변경 (코드 출력하지 않음)
-                        _history,
                         None,
                         gr.update(active_key="loading"),
                         gr.update(open=True)
@@ -310,7 +302,6 @@ class Demo:
                 async for content in try_openai_api(openai_messages):
                     yield [
                         "",  # 빈 문자열로 변경 (코드 출력하지 않음)
-                        _history,
                         None,
                         gr.update(active_key="loading"),
                         gr.update(open=True)
@@ -319,17 +310,9 @@ class Demo:
                     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=False)
@@ -342,9 +325,6 @@ class Demo:
             print(f"Error details: {str(e)}")
             raise ValueError(f'Error calling APIs: {str(e)}')
 
-    def clear_history(self):
-        return []
-
 def remove_code_block(text):
     pattern = r'```html\n(.+?)\n```'
     match = re.search(pattern, text, re.DOTALL)
@@ -482,6 +462,13 @@ def create_main_interface():
         custom_css = f.read()
 
     demo = gr.Blocks(css=custom_css + """
+        .empty-content {
+            padding: 40px !important;
+            background: #f8f9fa !important;
+            border-radius: 10px !important;
+            margin: 20px !important;
+        }    
+        
         .container {
             background: #f0f0f0;
             min-height: 100vh;
@@ -591,6 +578,20 @@ def create_main_interface():
                             gr.HTML("""
                                 <div class="thinking-container">
                                     <style>
+                                        .custom-textarea {
+                                            background: #f8f9fa !important;
+                                            border: 1px solid #e0e0e0 !important;
+                                            border-radius: 10px !important;
+                                            padding: 15px !important;
+                                            min-height: 150px !important;
+                                            font-family: -apple-system, BlinkMacSystemFont, sans-serif !important;
+                                        }
+        
+                                        .custom-textarea:focus {
+                                            border-color: #007aff !important;
+                                            box-shadow: 0 0 0 2px rgba(0,122,255,0.2) !important;
+                                        }
+        
                                         .thinking-container {
                                             text-align: center;
                                             padding: 20px;
@@ -756,14 +757,8 @@ def create_main_interface():
                                         size="large", 
                                         allow_clear=True,
                                         placeholder=get_random_placeholder(),
-                                        style={
-                                            "background": "#f8f9fa",
-                                            "border": "1px solid #e0e0e0",
-                                            "borderRadius": "10px",
-                                            "padding": "15px",
-                                            "minHeight": "150px",
-                                            "fontFamily": "-apple-system, BlinkMacSystemFont, sans-serif"
-                                        }
+                                        elem_classes="custom-textarea"  # style 대신 class 사용
+
                                     )
  
                                     # 버튼 그룹
@@ -772,33 +767,22 @@ def create_main_interface():
                                             "Generate", 
                                             type="primary", 
                                             size="large",
-                                            style={
-                                                "background": "#007aff",
-                                                "borderRadius": "8px",
-                                                "boxShadow": "0 2px 4px rgba(0,0,0,0.1)"
-                                            }
+                                            elem_classes="generate-btn"
                                         )
                                         boost_btn = antd.Button(
                                             "Enhance", 
                                             type="default", 
                                             size="large",
-                                            style={
-                                                "borderRadius": "8px",
-                                                "border": "1px solid #007aff",
-                                                "color": "#007aff"
-                                            }
+                                            elem_classes="enhance-btn"
                                         )
                                         deploy_btn = antd.Button(
                                             "Share", 
                                             type="default", 
                                             size="large",
-                                            style={
-                                                "borderRadius": "8px",
-                                                "border": "1px solid #28c840",
-                                                "color": "#28c840"
-                                            }
+                                            elem_classes="share-btn"
                                         )
 
+
                                     deploy_result = gr.HTML(
                                         label="Share Result",
                                         elem_classes="deploy-result"
@@ -826,13 +810,9 @@ def create_main_interface():
                                         with antd.Tabs.Item(key="empty"):
                                             empty = antd.Empty(
                                                 description="Enter your question to begin",
-                                                elem_classes="right_content",
-                                                style={
-                                                    "padding": "40px",
-                                                    "background": "#f8f9fa",
-                                                    "borderRadius": "10px",
-                                                    "margin": "20px"
-                                                }
+
+                                                elem_classes="right_content empty-content"  # style 대신 class 사용
+
                                             )
                                         
                                         with antd.Tabs.Item(key="loading"):
@@ -850,8 +830,8 @@ def create_main_interface():
 # 이벤트 핸들러 연결
                         btn.click(
                             demo_instance.generation_code,
-                            inputs=[input, setting, history],
-                            outputs=[code_output, history, sandbox, state_tab, code_drawer]
+                            inputs=[input, setting],
+                            outputs=[code_output, sandbox, state_tab, code_drawer]
                         ).then(  # then을 사용하여 연속 동작 추가
                             fn=update_placeholder,
                             inputs=[],
@@ -870,9 +850,39 @@ def create_main_interface():
                             outputs=[deploy_result]
                         )
 
-    # 추가 스타일 적용
     gr.HTML("""
         <style>
+            .generate-btn {
+                background: #007aff !important;
+                border-radius: 8px !important;
+                box-shadow: 0 2px 4px rgba(0,0,0,0.1) !important;
+            }
+            
+            .enhance-btn {
+                border-radius: 8px !important;
+                border: 1px solid #007aff !important;
+                color: #007aff !important;
+            }
+            
+            .share-btn {
+                border-radius: 8px !important;
+                border: 1px solid #28c840 !important;
+                color: #28c840 !important;
+            }
+            
+            /* hover 효과 */
+            .generate-btn:hover {
+                background: #0056b3 !important;
+            }
+            
+            .enhance-btn:hover {
+                background: rgba(0,122,255,0.1) !important;
+            }
+        
+            .share-btn:hover {
+                background: rgba(40,200,64,0.1) !important;
+            }
+
             .app-content {
                 padding: 20px;
                 text-align: center;