Update app.py

app.py

@@ -106,10 +106,12 @@ def interact(user_input, history):
 def send_selected_story(title, model_name, system_prompt):
     global chat_history
     global selected_story
+    global selected_system_prompt
     global data  # Ensure data is reset
     data = []  # Reset data for new story
     tokenizer, model = load_model(model_name)
     selected_story = title
+    selected_system_prompt = system_prompt
     for story in stories:
         if story["title"] == title:
             system_prompt = f"""
@@ -153,12 +155,14 @@ def save_comment_score(chat_responses, score, comment, story_name, user_name):
     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
+    system_prompt = selected_system_prompt
 
     # Append data to local data storage
     data.append([
         timestamp_str,
         user_name,
         model_name,
+        system_prompt,
         story_name,
         last_user_message,
         last_assistant_message,
@@ -168,9 +172,9 @@ def save_comment_score(chat_responses, score, comment, story_name, user_name):
 
     # Append data to Google Sheets
     sheet = client.open(google_sheets_name).worksheet(user_name)  # Save to user-specific sheet
-    sheet.append_row([timestamp_str, user_name, model_name, story_name, last_user_message, last_assistant_message, score, comment])
+    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", "Story Name", "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, gr.update(value="")  # Clear the comment input box
 
 # Create the chat interface using Gradio Blocks
@@ -200,7 +204,7 @@ with gr.Blocks() as demo:
             comment_input = gr.Textbox(placeholder="Add a comment...", label="Comment")
             save_button = gr.Button("Save Score and Comment")
 
-    data_table = gr.DataFrame(headers=["Timestamp", "User Name", "Model Name", "Story Name", "User Input", "Chat Response", "Score", "Comment"], visible=True)
+    data_table = gr.DataFrame(headers=["Timestamp", "User Name", "Model Name", "System Prompt", "Story Name", "User Input", "Chat Response", "Score", "Comment"], visible=True)
 
     chat_history_json = gr.JSON(value=[], visible=False)