Update app.py

app.py

@@ -214,10 +214,6 @@ def construct_input_prompt(chat_history, message, personalized_data):
     for user, assistant in chat_history:
         input_prompt += f"{user} [/INST] {assistant} <s>[INST] "
     input_prompt += f"{message} [/INST] "
-    print("FROM CONSTRUCT_INPUT_PROMPT")
-    print(f"THIS IS THE CHAT HISTORY: {chat_history}")
-    print(f"THIS IS THE MESSAGE: {message}")
-    print(f"THIS IS THE INPUT PROMPT: {input_prompt}")
     return input_prompt
 
 ## trust-game-llama-2-7b-chat
@@ -296,7 +292,6 @@ def generate(
     # Start the model generation thread
     t = Thread(target=model.generate, kwargs=generate_kwargs)
     t.start()
-    print(f"this is the thread: {t}")
 
     # Yield generated text chunks
     outputs = []
@@ -345,20 +340,24 @@ def generate(
         existing_data = pd.read_csv(data_file)
         
         # Add timestamp, message and a readable sentence column 
-        conversation_df = pd.DataFrame(conversation)
+        #conversation_df = pd.DataFrame(conversation)
+        conversation_df = pd.DataFrame()
         conversation_df['message'] = message
-        conversation_df['timestamp'] = timestamp
+        conversation_df['input_prompt'] = input_prompt
         conversation_df['readable_sentence'] = readable_sentence
-        
+        conversation_df['timestamp'] = timestamp
+
         # Append new conversation to existing data
         updated_data = pd.concat([existing_data, conversation_df], ignore_index=True)
         updated_data.to_csv(data_file, index=False)
     else:
         # If file doesn't exist, create new file with conversation data
-        conversation_df = pd.DataFrame(conversation)
+        #conversation_df = pd.DataFrame(conversation)
+        conversation_df = pd.DataFrame()
         conversation_df['message'] = message
-        conversation_df['timestamp'] = timestamp
+        conversation_df['input_prompt'] = input_prompt
         conversation_df['readable_sentence'] = readable_sentence
+        conversation_df['timestamp'] = timestamp
         conversation_df.to_csv(data_file, index=False)
 
     print("Updating .csv")