Update app.py

app.py

@@ -74,11 +74,12 @@ interaction_count = 0
 def interact(user_input, history):
     global tokenizer, model, interaction_count
     try:
-        if tokenizer is None or model is None:
+        if tokenizer is None or model is not None:
             raise ValueError("Tokenizer or model is not initialized.")
         
         # Increment interaction count
         interaction_count += 1
+        print(f"Interaction count: {interaction_count}")  # Print the interaction count
         
         # Check if the maximum number of interactions has been reached
         if interaction_count > MAX_INTERACTIONS:
@@ -123,8 +124,10 @@ def send_selected_story(title, model_name, system_prompt):
     interaction_count = 0  # Reset interaction count
     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:
@@ -141,7 +144,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=selected_story)  # Reset the data table and update the selected story textbox
+                return formatted_history, chat_history, gr.update(value=[]), gr.update(value=story_text)  # Reset the data table and update the selected story textbox
             else:
                 print("Combined message is empty.")
         else: