Update mouse.py

mouse.py

@@ -22,6 +22,7 @@ import string
 import random
 import requests  # 상단 import 구문에 추가
 
+__all__ = ['Demo', 'demo', 'SystemPrompt', 'mouse_interface', 'history_to_messages', 'messages_to_history']
 
 # SystemPrompt 부분을 직접 정의
 SystemPrompt = """너의 이름은 'MOUSE'이다. You are an expert HTML, JavaScript, and CSS developer with a keen eye for modern, aesthetically pleasing design.
@@ -901,12 +902,15 @@ def handle_boost(prompt: str):
         print(f"Boost 처리 중 오류: {str(e)}")
         return prompt, gr.update(active_key="empty")
 
-        
-# Demo 인스턴스 생성
-demo_instance = Demo()
 
+# 파일 상단에 한 번만 선언
+mouse_interface = None
 
+# with gr.Blocks 부분에서 설정
 with gr.Blocks(css_paths="app.css",theme=theme) as demo:
+    global mouse_interface
+    mouse_interface = demo
+ 
     history = gr.State([])
     setting = gr.State({
         "system": SystemPrompt,
@@ -1109,11 +1113,11 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
             )
 
 
-
+# mouse.py 파일 맨 아래의 실행 부분을 다음과 같이 수정
 if __name__ == "__main__":
     try:
         demo_instance = Demo()
-        demo.queue(default_concurrency_limit=20).launch(ssr_mode=False)
+        demo.queue(default_concurrency_limit=20)  # .launch() 부분만 제거
     except Exception as e:
         print(f"Initialization error: {e}")
         raise