Update app.py

app.py

@@ -13,8 +13,8 @@ torch.jit.script = lambda f: f
 # Initialize Google Sheets client
 client = init_google_sheets_client()
 sheet = client.open(google_sheets_name)
-stories_sheet = sheet.worksheet("Stories")  # Assuming stories are in the second sheet (index 1)
-system_prompts_sheet = sheet.worksheet("System Prompts")  # Assuming system prompts are in a separate sheet
+stories_sheet = sheet.worksheet("Stories")
+system_prompts_sheet = sheet.worksheet("System Prompts")
 
 # Load stories from Google Sheets
 def load_stories():
@@ -77,7 +77,6 @@ def interact(user_input, history, interaction_count):
             raise ValueError("Tokenizer or model is not initialized.")
         
         interaction_count += 1
-        print(f"Interaction count: {interaction_count}")
         
         messages = history + [{"role": "user", "content": user_input}]
         
@@ -93,7 +92,7 @@ def interact(user_input, history, interaction_count):
         
         # Generate response using selected model
         input_ids = tokenizer(prompt, return_tensors='pt').input_ids.to("cuda")
-        chat_history_ids = model.generate(input_ids, max_new_tokens=100, pad_token_id=tokenizer.eos_token_id)
+        chat_history_ids = model.generate(input_ids, max_new_tokens=100, pad_token_id=tokenizer.eos_token_id, temperature=0.1)
         response = tokenizer.decode(chat_history_ids[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
         
         # Update chat history with generated response
@@ -131,8 +130,6 @@ Here is the story:
             if combined_message:
                 chat_history = []  # Reset chat history
                 chat_history.append({"role": "system", "content": combined_message})
-
-                # Generate the first question based on the story
                 question_prompt = "Please ask a simple question about the story to encourage interaction."
                 _, formatted_history, chat_history, interaction_count = interact(question_prompt, chat_history, interaction_count)