Update app-backup.py

app-backup.py

@@ -232,8 +232,8 @@ def analyze_image_for_robot(
     prompt: str,
     task_type: str = "general",
     use_web_search: bool = False,
-    enable_thinking: bool = True,
-    max_new_tokens: int = 1024
+    enable_thinking: bool = False,  # 기본값 False로 변경
+    max_new_tokens: int = 250  # 기본값 250으로 변경
 ) -> str:
     """로봇 작업을 위한 이미지 분석"""
     global model, processor
@@ -247,23 +247,25 @@ def analyze_image_for_robot(
         if isinstance(image, np.ndarray):
             image = Image.fromarray(image).convert('RGB')
         
-        # 태스크별 시스템 프롬프트 구성
+        # 태스크별 시스템 프롬프트 구성 (더 간결하게)
         system_prompts = {
-            "general": "당신은 로봇 시각 시스템입니다. 이미지를 자세히 분석하고 설명하세요.",
-            "planning": """당신은 로봇 작업 계획을 수립하는 AI입니다. 
-주어진 이미지와 작업을 분석하여 단계별 실행 계획을 작성하세요.
-형식: Step_1: xxx\nStep_2: xxx\n...\nStep_n: xxx""",
-            "grounding": "당신은 객체 위치를 찾는 로봇 시각 시스템입니다. 요청된 객체의 위치를 [x1, y1, x2, y2] 좌표로 반환하세요.",
-            "affordance": "당신은 로봇 파지점을 분석하는 AI입니다. 주어진 작업을 위한 최적의 파지 영역을 [x1, y1, x2, y2] 좌표로 예측하세요.",
-            "trajectory": "당신은 로봇 경로를 계획하는 AI입니다. 목표 지점까지의 경로를 [(x1,y1), (x2,y2), ...] 형식으로 제시하세요.",
-            "pointing": "당신은 다중 지점을 지정하는 로봇 시각 시스템입니다. 요청된 위치들을 [(x1,y1), (x2,y2), ...] 형식으로 반환하세요."
+            "general": "당신은 로봇 시각 시스템입니다. 핵심 내용만 간결하게 설명하세요.",
+            "planning": """당신은 로봇 작업 계획 AI입니다. 
+주요 단계만 간결하게 작성하세요.
+형식: Step_1: xxx
+Step_2: xxx
+Step_n: xxx""",
+            "grounding": "당신은 객체 위치 시스템입니다. 객체 위치를 [x1, y1, x2, y2]로 반환하세요.",
+            "affordance": "당신은 파지점 분석 AI입니다. 파지 영역을 [x1, y1, x2, y2]로 반환하세요.",
+            "trajectory": "당신은 경로 계획 AI입니다. 경로를 [(x1,y1), (x2,y2), ...]로 제시하세요.",
+            "pointing": "당신은 지점 지정 시스템입니다. 위치를 [(x1,y1), (x2,y2), ...]로 반환하세요."
         }
         
         system_prompt = system_prompts.get(task_type, system_prompts["general"])
         
-        # Chain-of-Thought 추가
+        # Chain-of-Thought 추가 (선택적)
         if enable_thinking:
-            system_prompt += "\n\n추론 과정을 <thinking></thinking> 태그 안에 상세히 작성한 후 최종 답변을 제시하세요."
+            system_prompt += "\n\n추론 과정을 <thinking></thinking> 태그 안에 작성 후 최종 답변을 제시하세요."
         
         # 웹 검색 수행
         combined_system = system_prompt
@@ -468,6 +470,15 @@ css = """
     padding: 10px;
     background: #f8f9fa;
 }
+.auto-capture-status {
+    text-align: center;
+    padding: 5px;
+    border-radius: 5px;
+    margin: 5px 0;
+    font-weight: bold;
+    background: #e8f5e9;
+    color: #2e7d32;
+}
 """
 
 with gr.Blocks(title="🤖 로봇 시각 시스템 (Gemma3-4B)", css=css) as demo:
@@ -489,7 +500,7 @@ with gr.Blocks(title="🤖 로봇 시각 시스템 (Gemma3-4B)", css=css) as dem
             <li>🤏 로봇 파지점 분석 (Affordance)</li>
             <li>🛤️ 경로 계획 (Trajectory Planning)</li>
             <li>🔍 실시간 웹 검색 통합</li>
-            <li>📄 문서 분석 (PDF, CSV, TXT)</li>
+            <li>🔄 10초마다 자동 캡처 및 분석</li>
         </ul>
     </div>
     """)
@@ -508,6 +519,11 @@ with gr.Blocks(title="🤖 로봇 시각 시스템 (Gemma3-4B)", css=css) as dem
                     height=350
                 )
                 
+                # 자동 캡처 상태 표시
+                auto_capture_status = gr.HTML(
+                    '<div class="auto-capture-status">🔄 자동 캡처: 대기 중</div>'
+                )
+                
                 # 캡처된 이미지 표시
                 captured_image = gr.Image(
                     label="캡처된 이미지",
@@ -518,9 +534,16 @@ with gr.Blocks(title="🤖 로봇 시각 시스템 (Gemma3-4B)", css=css) as dem
             # 로봇 작업 버튼들
             gr.Markdown("### 🎯 로봇 작업 선택")
             with gr.Row():
-                capture_btn = gr.Button("📸 캡처", variant="primary", elem_classes="task-button")
+                capture_btn = gr.Button("📸 수동 캡처", variant="primary", elem_classes="task-button")
                 clear_capture_btn = gr.Button("🗑️ 초기화", elem_classes="task-button")
             
+            with gr.Row():
+                auto_capture_toggle = gr.Checkbox(
+                    label="🔄 자동 캡처 활성화 (10초마다)",
+                    value=False,
+                    info="활성화 시 10초마다 자동으로 캡처 및 분석"
+                )
+            
             with gr.Row():
                 planning_btn = gr.Button("📋 작업 계획", elem_classes="task-button")
                 grounding_btn = gr.Button("📍 객체 위치", elem_classes="task-button")
@@ -551,16 +574,16 @@ with gr.Blocks(title="🤖 로봇 시각 시스템 (Gemma3-4B)", css=css) as dem
                         
                         enable_thinking = gr.Checkbox(
                             label="🤔 추론 과정 표시",
-                            value=True,
+                            value=False,  # 기본값 False로 변경
                             info="Chain-of-Thought 추론 과정을 보여줍니다"
                         )
                     
                     max_tokens = gr.Slider(
                         label="최대 토큰 수",
-                        minimum=256,
+                        minimum=100,
                         maximum=4096,
-                        value=1024,
-                        step=256
+                        value=250,  # 기본값 250으로 변경
+                        step=50
                     )
             
             gr.Markdown("### 📊 분석 결과")
@@ -576,8 +599,8 @@ with gr.Blocks(title="🤖 로봇 시각 시스템 (Gemma3-4B)", css=css) as dem
                 '<div class="status-box" style="background:#d4edda; color:#155724;">🎮 시스템 준비 완료</div>'
             )
     
-    # 문서 분석 탭
-    with gr.Tab("📄 문서 분석"):
+    # 문서 분석 탭 (숨김 처리)
+    with gr.Tab("📄 문서 분석", visible=False):  # visible=False로 숨김
         with gr.Row():
             with gr.Column():
                 doc_files = gr.File(
@@ -609,6 +632,7 @@ with gr.Blocks(title="🤖 로봇 시각 시스템 (Gemma3-4B)", css=css) as dem
     
     # 이벤트 핸들러
     webcam_state = gr.State(None)
+    auto_capture_state = gr.State({"enabled": False, "timer": None})
     
     def capture_webcam(frame):
         """웹캠 프레임 캡처"""
@@ -636,29 +660,57 @@ with gr.Blocks(title="🤖 로봇 시각 시스템 (Gemma3-4B)", css=css) as dem
             max_new_tokens=tokens
         )
         
-        # 결과 포맷팅
+        # 결과 포맷팅 (더 간결하게)
         timestamp = time.strftime("%H:%M:%S")
         task_names = {
             "planning": "작업 계획",
-            "grounding": "객체 위치 파악",
-            "affordance": "파지점 분석",
+            "grounding": "객체 위치",
+            "affordance": "파지점",
             "trajectory": "경로 계획"
         }
         
-        formatted_result = f"""🤖 로봇 {task_names.get(task_type, '분석')} 결과:
-
-📸 **작업**: {prompt}
+        formatted_result = f"""🤖 {task_names.get(task_type, '분석')} 결과 ({timestamp}):
 
-📝 **분석 결과**:
-{result}
-
-⏰ 분석 시간: {timestamp}
-🎯 모델: {model_name}
-🔧 태스크: {task_type}"""
+{result}"""
         
         complete_status = '<div class="status-box" style="background:#d4edda; color:#155724;">✅ 분석 완료!</div>'
         return formatted_result, complete_status
     
+    # 자동 캡처 및 분석 함수
+    def auto_capture_and_analyze(webcam_frame, task_prompt, use_search, thinking, tokens, auto_state):
+        """자동 캡처 및 분석"""
+        if webcam_frame is None:
+            return (
+                None,
+                "자동 캡처 대기 중...",
+                '<div class="status-box" style="background:#fff3cd; color:#856404;">⏳ 웹캠 대기 중</div>',
+                '<div class="auto-capture-status">🔄 자동 캡처: 웹캠 대기 중</div>'
+            )
+        
+        # 캡처 수행
+        timestamp = time.strftime("%H:%M:%S")
+        
+        # 이미지 분석 (작업 계획 모드로)
+        result = analyze_image_for_robot(
+            image=webcam_frame,
+            prompt=task_prompt,
+            task_type="planning",
+            use_web_search=use_search,
+            enable_thinking=thinking,
+            max_new_tokens=tokens
+        )
+        
+        formatted_result = f"""🔄 자동 분석 ({timestamp}):
+
+{result}"""
+        
+        return (
+            webcam_frame,
+            formatted_result,
+            '<div class="status-box" style="background:#d4edda; color:#155724;">✅ 자동 분석 완료</div>',
+            f'<div class="auto-capture-status">🔄 자동 캡처: 마지막 분석 {timestamp}</div>'
+        )
+    
     # 웹캠 스트리밍
     webcam.stream(
         fn=lambda x: x,
@@ -666,7 +718,7 @@ with gr.Blocks(title="🤖 로봇 시각 시스템 (Gemma3-4B)", css=css) as dem
         outputs=[webcam_state]
     )
     
-    # 캡처 버튼
+    # 수동 캡처 버튼
     capture_btn.click(
         fn=capture_webcam,
         inputs=[webcam_state],
@@ -720,6 +772,29 @@ with gr.Blocks(title="🤖 로봇 시각 시스템 (Gemma3-4B)", css=css) as dem
         outputs=[doc_result]
     )
     
+    # 자동 캡처 타이머 (10초마다)
+    timer = gr.Timer(10.0, active=False)  # 10초 타이머, 초기에는 비활성화
+    
+    # 자동 캡처 토글 이벤트
+    def toggle_auto_capture(enabled):
+        if enabled:
+            return gr.Timer(10.0, active=True), '<div class="auto-capture-status">🔄 자동 캡처: 활성화됨 (10초마다)</div>'
+        else:
+            return gr.Timer(active=False), '<div class="auto-capture-status">🔄 자동 캡처: 비활성화됨</div>'
+    
+    auto_capture_toggle.change(
+        fn=toggle_auto_capture,
+        inputs=[auto_capture_toggle],
+        outputs=[timer, auto_capture_status]
+    )
+    
+    # 타이머 틱 이벤트
+    timer.tick(
+        fn=auto_capture_and_analyze,
+        inputs=[webcam_state, task_prompt, use_web_search, enable_thinking, max_tokens, auto_capture_state],
+        outputs=[captured_image, result_output, status_display, auto_capture_status]
+    )
+    
     # 초기 모델 로드
     def initial_load():
         load_model()