Update custom_utils.py

custom_utils.py

@@ -290,6 +290,8 @@ def handle_user_prompt(openai_api_key, prompt, db, collection, stages=[], vector
     search_results_df = pd.DataFrame([item.dict() for item in search_results_models])
 
     # Generate system response using OpenAI's completion
+    content = f"Answer this user question: {prompt} with the following context:\n{search_results_df}"
+    
     completion = openai.chat.completions.create(
         model="gpt-4o",
         messages=[
@@ -298,14 +300,17 @@ def handle_user_prompt(openai_api_key, prompt, db, collection, stages=[], vector
                 "content": "You are an AirBnB listing recommendation system."},
             {
                 "role": "user", 
-                "content": f"Answer this user question: {prompt} with the following context:\n{search_results_df}"
+                "content": content
             }
         ]
     )
 
     result = completion.choices[0].message.content
 
-    print(f"- Prompt:\n{prompt}\n")
-    print(f"- Completion:\n{result}\n")
+    print("###")
+    print(f"- User Content:\n{content}\n")
+    print("###")
+    print(f"- Prompt Completion:\n{result}\n")
+    print("###")
 
     return result