Update app.py

app.py

@@ -141,8 +141,15 @@ Here is the story:
 
 # Function to save comment and score
 def save_comment_score(chat_responses, score, comment, story_name, user_name, system_prompt):
-    chat_history_str = " ".join([f"{role}: {content}" for role, content in chat_responses])
-    
+    full_chat_history = ""
+
+    # Create formatted chat history with roles
+    for message in chat_responses:
+        if message[0]:  # User message
+            full_chat_history += f"User: {message[0]}\n"
+        if message[1]:  # Assistant message
+            full_chat_history += f"Assistant: {message[1]}\n"
+
     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
@@ -154,7 +161,7 @@ def save_comment_score(chat_responses, score, comment, story_name, user_name, sy
         model_name,
         system_prompt,
         story_name,
-        chat_history_str,
+        full_chat_history,
         score,
         comment
     ])
@@ -165,7 +172,7 @@ 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, chat_history_str, score, comment])
+    user_sheet.append_row([timestamp_str, user_name, model_name, system_prompt, story_name, full_chat_history, score, comment])
 
     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