Update app.py

app.py

@@ -686,6 +686,8 @@ def get_random_placeholder():
 def update_placeholder():
     return gr.update(placeholder=get_random_placeholder())
 
+
+
 def create_main_interface():
     """메인 인터페이스 생성 함수"""
     
@@ -852,18 +854,13 @@ def create_main_interface():
     """, theme=theme)
 
 
-
     with demo:
         with gr.Tabs(elem_classes="main-tabs") as tabs:
             with gr.Tab("Visual AI Assistant", elem_id="mouse-tab", elem_classes="mouse-tab"):
-                # SystemPrompt 설정을 위한 State 추가
-                setting = gr.State({
-                    "system": SystemPrompt,
-                })
+                setting = gr.State({"system": SystemPrompt})
                 
                 with ms.Application() as app:
                     with antd.ConfigProvider():
-
                         with antd.Drawer(open=False, title="AI is Creating...", placement="left", width="750px") as code_drawer:
                             gr.HTML("""
                                 <div class="thinking-container">
@@ -1012,15 +1009,11 @@ def create_main_interface():
                             """)
                             code_output = legacy.Markdown(visible=False)
 
-
-# 메인 컨텐츠를 위한 Row
                         with antd.Row(gutter=[32, 12]) as layout:    
                             # 좌측 패널
                             with antd.Col(span=24, md=8):
                                 with antd.Flex(vertical=True, gap="middle", wrap=True):
-
-
-        
+                                    # macOS 스타일 윈도우 헤더
                                     # macOS 스타일 윈도우 헤더
                                     header = gr.HTML("""
                                         <div class="window-frame">
@@ -1046,29 +1039,31 @@ Use the "Generate" button for basic creation, "Enhance" button for prompt improv
                                         </div>
                                     """.format(get_image_base64('mouse.gif')))
 
-        # 파일 업로드 섹션
-                                    file_upload = gr.File(
-                                        label="Upload Files (Images/Documents)",
-                                        file_types=[".jpg", ".jpeg", ".png", ".gif", ".bmp", ".webp", 
-                                                  ".txt", ".pdf", ".docx", ".rtf"],
-                                        file_count="multiple",
-                                        elem_classes="file-upload"
-                                    )
-        
-                                    file_analysis = gr.Textbox(
-                                        label="File Analysis Result",
-                                        interactive=False,
-                                        elem_classes="analysis-result"
-                                    )
-        
-                                    analyze_btn = antd.Button(
-                                        "Analyze Files",
-                                        type="primary",
-                                        size="large",
-                                        elem_classes="analyze-btn"
-                                    )
 
-        # 입력 영역
+                                    # 파일 업로드 섹션
+                                    with gr.Box(elem_classes="file-upload-section"):
+                                        file_upload = gr.File(
+                                            label="Upload Files (Images/Documents)",
+                                            file_types=[".jpg", ".jpeg", ".png", ".gif", ".bmp", ".webp", 
+                                                      ".txt", ".pdf", ".docx", ".rtf"],
+                                            file_count="multiple",
+                                            elem_classes="file-upload"
+                                        )
+                                        
+                                        file_analysis = gr.Textbox(
+                                            label="File Analysis Result",
+                                            interactive=False,
+                                            elem_classes="analysis-result"
+                                        )
+                                        
+                                        analyze_btn = antd.Button(
+                                            "Analyze Files",
+                                            type="primary",
+                                            size="large",
+                                            elem_classes="analyze-btn"
+                                        )
+
+                                    # 입력 영역
                                     input = antd.InputTextarea(
                                         size="large", 
                                         allow_clear=True,
@@ -1076,7 +1071,7 @@ Use the "Generate" button for basic creation, "Enhance" button for prompt improv
                                         elem_classes="custom-textarea"
                                     )
 
-        # 버튼 그룹
+                                    # 버튼 그룹
                                     with antd.Flex(gap="small", justify="flex-start"):
                                         btn = antd.Button(
                                             "Generate", 
@@ -1097,68 +1092,14 @@ Use the "Generate" button for basic creation, "Enhance" button for prompt improv
                                             elem_classes="share-btn"
                                         )
 
-                                        deploy_result = gr.HTML(
-                                            label="Share Result",
-                                            elem_classes="deploy-result"
-                                        )
-
-
-# 이벤트 핸들러들을 한 곳에 모아서 정의
-                        async def handle_file_upload(files):
-                            if not files:
-                                return "No files uploaded"
-    
-                            if len(files) > FileProcessor.MAX_FILES:
-                                return f"Maximum {FileProcessor.MAX_FILES} files allowed"
-    
-                            results = []
-                            for file in files:
-                                if not FileProcessor.is_allowed_file(file.name):
-                                    results.append(f"Unsupported file: {file.name}")
-                                    continue
-        
-                                if os.path.getsize(file.name) > FileProcessor.MAX_FILE_SIZE:
-                                    results.append(f"File too large: {file.name}")
-                                    continue
-        
-                                result = await FileProcessor.process_file(file)
-                                results.append(f"Analysis for {file.name}:\n{result}\n")
-    
-                            return "\n".join(results)
-
-# 이벤트 핸들러 연결
-                        analyze_btn.click(
-                            fn=handle_file_upload,
-                            inputs=[file_upload],
-                            outputs=[file_analysis]
-                        )
-
-                        btn.click(
-                            demo_instance.generation_code,
-                            inputs=[input, setting],
-                            outputs=[code_output, sandbox, state_tab, code_drawer]
-                        ).then(
-                            fn=update_placeholder,
-                            inputs=[],
-                            outputs=[input]
-                        )
-
-                        boost_btn.click(
-                            fn=handle_boost,
-                            inputs=[input],
-                            outputs=[input, state_tab]
-                        )
-
-                        deploy_btn.click(
-                            fn=lambda code: deploy_to_vercel(remove_code_block(code)) if code else "No code to share.",
-                            inputs=[code_output],
-                            outputs=[deploy_result]
-                        )
+                                    deploy_result = gr.HTML(
+                                        label="Share Result",
+                                        elem_classes="deploy-result"
+                                    )
 
                             # 우측 패널
                             with antd.Col(span=24, md=16):
                                 with ms.Div(elem_classes="right_panel"):
-                                    # macOS 스타일 윈도우 헤더
                                     gr.HTML("""
                                         <div class="window-frame">
                                             <div class="window-header">
@@ -1172,14 +1113,11 @@ Use the "Generate" button for basic creation, "Enhance" button for prompt improv
                                         </div>
                                     """)
 
-                                    # 탭 컨텐츠
                                     with antd.Tabs(active_key="empty", render_tab_bar="() => null") as state_tab:
                                         with antd.Tabs.Item(key="empty"):
                                             empty = antd.Empty(
                                                 description="Enter your question to begin",
-
-                                                elem_classes="right_content empty-content"  # style 대신 class 사용
-
+                                                elem_classes="right_content empty-content"
                                             )
                                         
                                         with antd.Tabs.Item(key="loading"):
@@ -1192,11 +1130,40 @@ Use the "Generate" button for basic creation, "Enhance" button for prompt improv
                                         
                                         with antd.Tabs.Item(key="render"):
                                             sandbox = gr.HTML(elem_classes="html_content")
+
+                        # 이벤트 핸들러 정의
+                        async def handle_file_upload(files):
+                            if not files:
+                                return "No files uploaded"
                             
+                            if len(files) > FileProcessor.MAX_FILES:
+                                return f"Maximum {FileProcessor.MAX_FILES} files allowed"
+                            
+                            results = []
+                            for file in files:
+                                if not FileProcessor.is_allowed_file(file.name):
+                                    results.append(f"Unsupported file: {file.name}")
+                                    continue
+                                
+                                if os.path.getsize(file.name) > FileProcessor.MAX_FILE_SIZE:
+                                    results.append(f"File too large: {file.name}")
+                                    continue
+                                
+                                result = await FileProcessor.process_file(file)
+                                results.append(f"Analysis for {file.name}:\n{result}\n")
+                            
+                            return "\n".join(results)
+
+                        # 이벤트 핸들러 연결
+                        analyze_btn.click(
+                            fn=handle_file_upload,
+                            inputs=[file_upload],
+                            outputs=[file_analysis]
+                        )
 
                         btn.click(
                             demo_instance.generation_code,
-                            inputs=[input, setting],  # setting이 이제 정의됨
+                            inputs=[input, setting],
                             outputs=[code_output, sandbox, state_tab, code_drawer]
                         ).then(
                             fn=update_placeholder,
@@ -1204,7 +1171,6 @@ Use the "Generate" button for basic creation, "Enhance" button for prompt improv
                             outputs=[input]
                         )
 
-
                         boost_btn.click(
                             fn=handle_boost,
                             inputs=[input],
@@ -1216,6 +1182,7 @@ Use the "Generate" button for basic creation, "Enhance" button for prompt improv
                             inputs=[code_output],
                             outputs=[deploy_result]
                         )
+                        
 
     gr.HTML("""
         <style>
@@ -1305,6 +1272,9 @@ Use the "Generate" button for basic creation, "Enhance" button for prompt improv
 
     return demo
 
+
+
+
 # 메인 실행 부분
 if __name__ == "__main__":
     try: