Update app.py

app.py

@@ -25,7 +25,7 @@ def load_stories():
 # Load system prompts from Google Sheets
 def load_system_prompts():
     system_prompts_data = system_prompts_sheet.get_all_values()
-    system_prompts = [prompt[0] for prompt in system_prompts_data if prompt[0] != "System Prompt"]  # Skip header row
+    system_prompts = [prompt[0] for prompt in system_prompts_data[1:]]  # Skip header row
     return system_prompts
 
 # Load available stories and system prompts
@@ -168,8 +168,12 @@ def save_comment_score(chat_responses, score, comment, story_name, user_name, sy
     ])
 
     # Append data to Google Sheets
-    sheet = client.open(google_sheets_name).sheet1  # Assuming results are saved in sheet1
-    sheet.append_row([timestamp_str, user_name, model_name, system_prompt, story_name, last_user_message, last_assistant_message, score, comment])
+    try:
+        user_sheet = client.open(google_sheets_name).worksheet(user_name)
+    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])
 
     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