Update app.py

app.py

@@ -11,6 +11,7 @@ from PIL import Image
 from io import BytesIO
 import base64
 import time
+from mouse import demo as mouse_demo  # mouse.py의 demo 객체를 import
 
 def take_screenshot(url):
     """웹사이트 스크린샷 촬영 함수 (로딩 대기 시간 추가)"""
@@ -825,14 +826,45 @@ def get_user_spaces():
             <p>Please try again later.</p>
         </div>
         """
-
         
-# Creating the Gradio interface
-demo = gr.Blocks()
+def create_main_interface():
+    """메인 인터페이스 생성 함수"""
+    
+    # 갤러리 탭용 Blocks 객체 생성
+    gallery_tab = gr.Blocks()
+    with gallery_tab:
+        gr.HTML(value=get_user_spaces())
+
+    # 전체 인터페이스를 Blocks으로 감싸고 Tabs 추가
+    demo = gr.Blocks(css="""
+        .main-tabs > div.tab-nav > button {
+            font-size: 1.1em !important;
+            padding: 0.5em 1em !important;
+            background: rgba(255, 255, 255, 0.8) !important;
+            border: none !important;
+            border-radius: 8px 8px 0 0 !important;
+            margin-right: 4px !important;
+        }
+        .main-tabs > div.tab-nav > button.selected {
+            background: linear-gradient(45deg, #0084ff, #00a3ff) !important;
+            color: white !important;
+        }
+        .main-tabs {
+            margin-top: -20px !important;
+            border-radius: 0 0 15px 15px !important;
+            box-shadow: 0 4px 15px rgba(0,0,0,0.1) !important;
+        }
+    """)
+    
+    with demo:
+        with gr.Tabs(elem_classes="main-tabs") as tabs:
+            with gr.Tab("갤러리", elem_id="gallery-tab"):
+                gr.HTML(value=get_user_spaces())
+            
+            with gr.Tab("MOUSE", elem_id="mouse-tab"):
+                # mouse.py의 인터페이스를 여기에 포함
+                mouse_demo.render()
 
-with demo:
-    html_output = gr.HTML(value=get_user_spaces())
+    return demo
 
-if __name__ == "__main__":
-    demo.launch()