Update app.py

app.py

@@ -13,7 +13,7 @@ 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)
+stories_sheet = sheet.worksheet("Stories")  # Fetch stories from the "Stories" sheet
 
 # Load stories from Google Sheets
 def load_stories():
@@ -102,7 +102,6 @@ def send_selected_story(title, model_name, system_prompt):
     data = []  # Reset data for new story
     tokenizer, model = load_model(model_name)
     selected_story = title
-    story_text = ""
     for story in stories:
         if story["title"] == title:
             system_prompt = f"""
@@ -112,7 +111,6 @@ Here is the story:
 {story['story']}
 ---
             """
-            story_text = story["story"]
             combined_message = system_prompt.strip()
             if combined_message:
                 chat_history = []  # Reset chat history
@@ -122,7 +120,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=story_text), gr.update(value=[])  # Reset the data table
+                return formatted_history, chat_history, gr.update(value=[]), story['story']  # Reset the data table and return the story content
             else:
                 print("Combined message is empty.")
         else:
@@ -160,8 +158,8 @@ def save_comment_score(chat_responses, score, comment, story_name, user_name):
         comment
     ])
 
-    # Select the sheet corresponding to the user name
-    user_sheet = client.open(google_sheets_name).worksheet(user_name)  # Assuming each user has a separate sheet named after them
+    # Append data to the user-specific sheet
+    user_sheet = client.open(google_sheets_name).worksheet(user_name)  # Open the sheet with the user's name
     user_sheet.append_row([timestamp_str, user_name, model_name, story_name, last_user_message, last_assistant_message, score, comment])
 
     df = pd.DataFrame(data, columns=["Timestamp", "User Name", "Model Name", "Story Name", "User Input", "Chat Response", "Score", "Comment"])
@@ -183,7 +181,6 @@ with gr.Blocks() as demo:
     system_prompt_input = gr.Textbox(lines=5, value=default_system_prompt, label="System Prompt")
 
     send_story_button = gr.Button("Send Story")
-
     selected_story_output = gr.Textbox(label="Selected Story", interactive=False)
 
     with gr.Row():
@@ -204,7 +201,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_input], outputs=[chatbot_output, chat_history_json, selected_story_output, data_table])
+    send_story_button.click(fn=send_selected_story, inputs=[story_dropdown, model_dropdown, system_prompt_input], outputs=[chatbot_output, chat_history_json, data_table, selected_story_output])
     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], outputs=[data_table, comment_input])