Update app.py

app.py

@@ -13,8 +13,8 @@ torch.jit.script = lambda f: f
 # Initialize Google Sheets client
 client = init_google_sheets_client()
 sheet = client.open(google_sheets_name)
-stories_sheet = sheet.get_worksheet(1)  # Assuming stories are in the second sheet (index 1)
-prompts_sheet = sheet.get_worksheet(2)  # Assuming system prompts are in the third sheet (index 2)
+stories_sheet = sheet.worksheet("Stories")  # Assuming stories are in the second sheet (index 1)
+system_prompts_sheet = sheet.worksheet("System Prompts")  # Assuming system prompts are in a separate sheet
 
 # Load stories from Google Sheets
 def load_stories():
@@ -22,16 +22,14 @@ def load_stories():
     stories = [{"title": story[0], "story": story[1]} for story in stories_data if story[0] != "Title"]  # Skip header row
     return stories
 
-# Load available stories
-stories = load_stories()
-
 # Load system prompts from Google Sheets
 def load_system_prompts():
-    prompts_data = prompts_sheet.get_all_values()
-    system_prompts = [row[0] for row in prompts_data if row[0] != "System Prompt"]  # Skip header row
+    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
     return system_prompts
 
-# Load available system prompts
+# Load available stories and system prompts
+stories = load_stories()
 system_prompts = load_system_prompts()
 
 # Initialize the selected model
@@ -130,7 +128,7 @@ Here is the story:
                 question_prompt = "Please ask a simple question about the story to encourage interaction."
                 _, formatted_history, chat_history = interact(question_prompt, chat_history)
 
-                return formatted_history, chat_history, gr.update(value=[]), gr.update(value=story['story'])  # Reset the data table and update selected story
+                return formatted_history, chat_history, gr.update(value=[]), story["story"]  # Reset the data table and return the story
             else:
                 print("Combined message is empty.")
         else:
@@ -173,12 +171,8 @@ def save_comment_score(chat_responses, score, comment, story_name, user_name, sy
     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])
 
-    # Create a DataFrame for display
-    display_data = [[last_user_message, last_assistant_message, score, comment]]
-    df_display = pd.DataFrame(display_data, columns=["User Input", "Chat Response", "Score", "Comment"])
-
     df = pd.DataFrame(data, columns=["Timestamp", "User Name", "Model Name", "System Prompt", "Story Name", "User Input", "Chat Response", "Score", "Comment"])
-    return df_display, gr.update(value="")  # Clear the comment input box
+    return df[["User Input", "Chat Response", "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:
@@ -188,10 +182,11 @@ with gr.Blocks() as demo:
     user_dropdown = gr.Dropdown(choices=user_names, label="Select User Name")
     initial_story = stories[0]["title"] if stories else None
     story_dropdown = gr.Dropdown(choices=[story["title"] for story in stories], label="Select Story", value=initial_story)
-    system_prompt_dropdown = gr.Dropdown(choices=system_prompts, label="Select System Prompt", value=system_prompts[0] if system_prompts else "")
+
+    system_prompt_dropdown = gr.Dropdown(choices=system_prompts, label="Select System Prompt")
 
     send_story_button = gr.Button("Send Story")
-    selected_story_display = gr.Textbox(label="Selected Story", interactive=False)
+    selected_story_textbox = gr.Textbox(label="Selected Story", lines=10, interactive=False)
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -211,7 +206,7 @@ with gr.Blocks() as demo:
 
     chat_history_json = gr.JSON(value=[], visible=False)
 
-    send_story_button.click(fn=send_selected_story, inputs=[story_dropdown, model_dropdown, system_prompt_dropdown], outputs=[chatbot_output, chat_history_json, data_table, selected_story_display])
+    send_story_button.click(fn=send_selected_story, inputs=[story_dropdown, model_dropdown, system_prompt_dropdown], outputs=[chatbot_output, chat_history_json, data_table, selected_story_textbox])
     send_message_button.click(fn=interact, inputs=[chatbot_input, chat_history_json], outputs=[chatbot_input, chatbot_output, chat_history_json])
     save_button.click(fn=save_comment_score, inputs=[chatbot_output, score_input, comment_input, story_dropdown, user_dropdown, system_prompt_dropdown], outputs=[data_table, comment_input])