Update app.py

app.py

@@ -864,58 +864,58 @@ with gr.Blocks(fill_height=True, title="Enhanced ThinkFlow") as demo:
                 temperature = gr.Slider(0.1, 1.0, 0.7, step=0.1, label="온도")
                 memory_weight = gr.Slider(0.0, 1.0, 0.5, step=0.1, label="메모리 반영 가중치")
 
-# 피드백 처리 함수
-     def process_positive_feedback():
-          global buffer_manager, current_contexts
-          if buffer_manager:
-               buffer_manager.update_retrieval_reward(current_contexts, reward=1.0)
-          return "피드백 감사합니다! 이 접근 방식을 향후 유사한 질문에 더 자주 사용하겠습니다."
-        
-     def process_negative_feedback():
-          global buffer_manager, current_contexts
-          if buffer_manager:
-               buffer_manager.update_retrieval_reward(current_contexts, reward=-0.5)
-          return "피드백 감사합니다! 이 접근 방식을 개선하겠습니다."
-    
-     def clear_memory():
-          global buffer_manager
-          if buffer_manager:
-               buffer_manager.clear()
-          return "메모리가 초기화되었습니다."
-    
-     def update_memory_displays():
-          global buffer_manager
-          if not buffer_manager:
-               return "메모리가 초기화되지 않았습니다.", "그래프가 초기화되지 않았습니다."
+    # 피드백 처리 함수
+    def process_positive_feedback():
+        global buffer_manager, current_contexts
+        if buffer_manager:
+            buffer_manager.update_retrieval_reward(current_contexts, reward=1.0)
+        return "피드백 감사합니다! 이 접근 방식을 향후 유사한 질문에 더 자주 사용하겠습니다."
+    
+    def process_negative_feedback():
+        global buffer_manager, current_contexts
+        if buffer_manager:
+            buffer_manager.update_retrieval_reward(current_contexts, reward=-0.5)
+        return "피드백 감사합니다! 이 접근 방식을 개선하겠습니다."
+    
+    def clear_memory():
+        global buffer_manager
+        if buffer_manager:
+            buffer_manager.clear()
+        return "메모리가 초기화되었습니다."
+    
+    def update_memory_displays():
+        global buffer_manager
+        if not buffer_manager:
+            return "메모리가 초기화되지 않았습니다.", "그래프가 초기화되지 않았습니다."
             
-          semantic_text = "현재 저장된 메모리:\n\n"
-          for i, mem in enumerate(buffer_manager.semantic_memory.memories[:5]):  # 최대 5개만 표시
-               semantic_text += f"{i+1}. {mem['text'][:100]}...\n\n"
+        semantic_text = "현재 저장된 메모리:\n\n"
+        for i, mem in enumerate(buffer_manager.semantic_memory.memories[:5]):  # 최대 5개만 표시
+            semantic_text += f"{i+1}. {mem['text'][:100]}...\n\n"
             
-          graph_text = "현재 그래프 노드:\n\n"
-          for node in buffer_manager.graph_memory.graph.nodes():
-               node_text = buffer_manager.graph_memory.get_text_by_node(node)
-               neighbors = list(buffer_manager.graph_memory.graph.neighbors(node))
-               graph_text += f"노드: {node}\n설명: {node_text[:50]}...\n연결: {', '.join(neighbors[:3])}\n\n"
+        graph_text = "현재 그래프 노드:\n\n"
+        for node in buffer_manager.graph_memory.graph.nodes():
+            node_text = buffer_manager.graph_memory.get_text_by_node(node)
+            neighbors = list(buffer_manager.graph_memory.graph.neighbors(node))
+            graph_text += f"노드: {node}\n설명: {node_text[:50]}...\n연결: {', '.join(neighbors[:3])}\n\n"
             
-          return semantic_text, graph_text
-    
-     # 초기화 함수
-     def initialize_models():
-          global pipe, buffer_manager, model_name
-          try:
-               pipe, buffer_manager = initialize_model_and_manager(model_name)
-               semantic_text, graph_text = update_memory_displays()
-               return "모델이 초기화되었습니다.", semantic_text, graph_text
-          except Exception as e:
-               return f"모델 초기화 오류: {str(e)}", "", ""
-    
-     # 모델 선택 변경 시 처리
-     def change_model(new_model_name):
-          global model_name
-          model_name = new_model_name
-          status, semantic_text, graph_text = initialize_models()
-          return status, semantic_text, graph_text
+        return semantic_text, graph_text
+    
+    # 초기화 함수
+    def initialize_models():
+        global pipe, buffer_manager, model_name
+        try:
+            pipe, buffer_manager = initialize_model_and_manager(model_name)
+            semantic_text, graph_text = update_memory_displays()
+            return "모델이 초기화되었습니다.", semantic_text, graph_text
+        except Exception as e:
+            return f"모델 초기화 오류: {str(e)}", "", ""
+    
+    # 모델 선택 변경 시 처리
+    def change_model(new_model_name):
+        global model_name
+        model_name = new_model_name
+        status, semantic_text, graph_text = initialize_models()
+        return status, semantic_text, graph_text