Update app.py

app.py

@@ -102,6 +102,7 @@ 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"""
@@ -111,6 +112,7 @@ Here is the story:
 {story['story']}
 ---
             """
+            story_text = story["story"]
             combined_message = system_prompt.strip()
             if combined_message:
                 chat_history = []  # Reset chat history
@@ -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=selected_story), gr.update(value=[])  # Reset the data table
+                return formatted_history, chat_history, gr.update(value=story_text), gr.update(value=[])  # Reset the data table
             else:
                 print("Combined message is empty.")
         else: