debug

core/launch_gradio.py

@@ -24,9 +24,9 @@ def create_interface(ctx: ContextManager, makePipeline: MakePipeline):
         def render_chat(ctx: ContextManager):
             html = ""
             for item in ctx.getHistory():
-                if item["role"] == ctx.getUserName():
+                if item["role"] == "user":
                     html += f"<div class='bubble-right'>{item['text']}</div>"
-                elif item["role"] == ctx.getBotName():
+                elif item["role"] == "bot":
                     html += f"<div class='bubble-left'>{item['text']}</div>"
             return gr.update(value=html)
 

core/make_reply.py

@@ -23,6 +23,7 @@ def generate_valid_response(ctx, makePipeline, user_msg) -> str:
         prompt = build_prompt(ctx.getHistory(), user_msg, user_name, bot_name)
         full_text = makePipeline.character_chat(prompt)
         response = extract_response(full_text)
+        print(f"debug: {response}")
         if is_valid_response(response, user_name, bot_name):
             break
     return clean_response(response, bot_name)
@@ -62,8 +63,6 @@ def extract_response(full_text):
 def is_valid_response(text: str, user_name, bot_name) -> bool:
     if user_name + ":" in text:
         return False
-    if bot_name + ":" in text:
-        return False
     return True
 
 # 응답 형식 정리