Update app.py

app.py

@@ -130,6 +130,11 @@ def generate(
     params = request.query_params
     print('those are the query params')
     print(params)
+
+    if request:
+        print("Request headers dictionary:", request.headers)
+        print("IP address:", request.client.host)
+        print("Query parameters:", dict(request.query_params))
     
     # Construct the input prompt using the functions from the system_prompt_config module
     input_prompt = construct_input_prompt(chat_history, message)
@@ -177,6 +182,13 @@ def generate(
         outputs.append(text)
         yield "".join(outputs)
 
+    # Fix bug that last answer is not recorded! 
+    # Parse the outputs into a readable sentence and record them
+    # Filter out empty strings and join the remaining strings with spaces
+    readable_sentence = ' '.join(filter(lambda x: x.strip(), outputs))
+    # Print the readable sentence
+    print(readable_sentence)
+
     # Save chat history to .csv file on HuggingFace Hub 
     # Generate filename with bot id and session id
     filename = f"{DATA_FILENAME}"
@@ -192,7 +204,8 @@ def generate(
         
         # Add timestamp column
         conversation_df = pd.DataFrame(conversation)
-        #conversation_df['readable_sentence'] = readable_sentence
+        conversation_df['ip_address'] = request.client.host
+        conversation_df['readable_sentence'] = readable_sentence
         conversation_df['timestamp'] = timestamp
         
         # Append new conversation to existing data
@@ -201,7 +214,8 @@ def generate(
     else:
         # If file doesn't exist, create new file with conversation data
         conversation_df = pd.DataFrame(conversation)
-        #conversation_df['readable_sentence'] = readable_sentence
+        conversation_df['ip_address'] = request.client.host
+        conversation_df['readable_sentence'] = readable_sentence
         conversation_df['timestamp'] = timestamp
         conversation_df.to_csv(data_file, index=False)