Update app.py

app.py

@@ -141,20 +141,8 @@ Here is the story:
 
 # Function to save comment and score
 def save_comment_score(chat_responses, score, comment, story_name, user_name, system_prompt):
-    last_user_message = ""
-    last_assistant_message = ""
-
-    # Find the last user and assistant messages
-    for message in reversed(chat_responses):
-        if isinstance(message, list) and len(message) == 2:
-            if message[0] and not last_user_message:
-                last_user_message = message[0]
-            elif message[1] and not last_assistant_message:
-                last_assistant_message = message[1]
-        
-        if last_user_message and last_assistant_message:
-            break
-
+    chat_history_str = " ".join([f"{role}: {content}" for role, content in chat_responses])
+    
     timestamp = datetime.now(timezone.utc) - timedelta(hours=3)  # Adjust to GMT-3
     timestamp_str = timestamp.strftime("%Y-%m-%d %H:%M:%S")
     model_name = selected_model
@@ -166,8 +154,7 @@ def save_comment_score(chat_responses, score, comment, story_name, user_name, sy
         model_name,
         system_prompt,
         story_name,
-        last_user_message,
-        last_assistant_message,
+        chat_history_str,
         score,
         comment
     ])
@@ -178,10 +165,10 @@ def save_comment_score(chat_responses, score, comment, story_name, user_name, sy
     except gspread.exceptions.WorksheetNotFound:
         user_sheet = client.open(google_sheets_name).add_worksheet(title=user_name, rows="100", cols="20")
     
-    user_sheet.append_row([timestamp_str, user_name, model_name, system_prompt, story_name, last_user_message, last_assistant_message, score, comment])
+    user_sheet.append_row([timestamp_str, user_name, model_name, system_prompt, story_name, chat_history_str, score, comment])
 
-    df = pd.DataFrame(data, columns=["Timestamp", "User Name", "Model Name", "System Prompt", "Story Name", "User Input", "Chat Response", "Score", "Comment"])
-    return df[["User Input", "Chat Response", "Score", "Comment"]], gr.update(value="")  # Show only the required columns and clear the comment input box
+    df = pd.DataFrame(data, columns=["Timestamp", "User Name", "Model Name", "System Prompt", "Story Name", "Chat History", "Score", "Comment"])
+    return df[["Chat History", "Score", "Comment"]], gr.update(value="")  # Show only the required columns and clear the comment input box
 
 # Create the chat interface using Gradio Blocks
 with gr.Blocks() as demo:
@@ -210,7 +197,7 @@ with gr.Blocks() as demo:
                 score_input = gr.Slider(minimum=0, maximum=5, step=1, label="Score")
                 comment_input = gr.Textbox(placeholder="Add a comment...", label="Comment")
                 save_button = gr.Button("Save Score and Comment")
-                data_table = gr.DataFrame(headers=["User Input", "Chat Response", "Score", "Comment"])
+                data_table = gr.DataFrame(headers=["Chat History", "Score", "Comment"])
 
         with gr.TabItem("User Guide"):
             gr.Textbox(label="User Guide", value='''