Update app.py

app.py

@@ -102,13 +102,15 @@ 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:
+            story_text = story['story']
             system_prompt = f"""
 {system_prompt}
 Here is the story:
 ---
-{story['story']}
+{story_text}
 ---
             """
             combined_message = system_prompt.strip()
@@ -120,7 +122,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=[])  # Reset the data table
+                return formatted_history, chat_history, story_text, gr.update(value=[])  # Reset the data table
             else:
                 print("Combined message is empty.")
         else:
@@ -190,6 +192,8 @@ with gr.Blocks() as demo:
         with gr.Column(scale=2):
             chatbot_output = gr.Chatbot(label="Chat History")
 
+    story_display = gr.Textbox(label="Selected Story", lines=10, interactive=False)
+
     with gr.Row():
         with gr.Column(scale=1):
             score_input = gr.Slider(minimum=0, maximum=5, step=1, label="Score")
@@ -200,7 +204,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, data_table])
+    send_story_button.click(fn=send_selected_story, inputs=[story_dropdown, model_dropdown, system_prompt_input], outputs=[chatbot_output, chat_history_json, story_display, data_table])
     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])