Update app.py

app.py

@@ -78,34 +78,25 @@ def get_random_row_from_dataset():
     # 유효한 행들 중 2개를 랜덤하게 선택
     chosen_rows = valid_rows.sample(2)
     
-    # 두 행의 text를 결합하여 하나의 대화 텍스트로 만듭니다.
-    combined_text = f"{chosen_rows.iloc[0]['text'].strip()} [turn] {chosen_rows.iloc[1]['text'].strip()}"
-    
-    return {"text": combined_text}
+    return chosen_rows[0], chosen_rows[1]
 
 def get_conversation():
     """
     get_random_row_from_dataset()를 호출하여 대화 문자열을 가져오고,
     "[turn]" 구분자를 기준으로 인간 메시지와 AI 메시지를 분리하여 반환합니다.
     """
-    row = get_random_row_from_dataset()
+    row1, row2 = get_random_row_from_dataset()
     if row is None:
         return "No valid conversation available.", "No valid conversation available."
     else:
-        raw_text = row['text']
-        parts = raw_text.split("[turn]")
-        if len(parts) < 2:
-            return "Invalid conversation format", "Invalid conversation format"
-        human_message = parts[0].strip()
-        ai_message = parts[1].strip()
-        return human_message, ai_message
+        return row1['text'], row2['text']
 
 # Gradio 인터페이스 생성 (왼쪽: Human Message, 오른쪽: AI Message)
 with gr.Blocks() as demo:
     gr.Markdown("## Random Conversation Generator")
     with gr.Row():
-        human_text = gr.Textbox(label="Human Message", lines=10, interactive=False)
-        ai_text = gr.Textbox(label="AI Message", lines=10, interactive=False)
+        conversation_1 = gr.Textbox(label="Conversation1", lines=10, interactive=False)
+        conversation_2 = gr.Textbox(label="Conversation2", lines=10, interactive=False)
     generate_btn = gr.Button("Generate Conversation")
     generate_btn.click(fn=get_conversation, inputs=[], outputs=[human_text, ai_text])