Rename openai_embedding.py to openai_.py

openai_embedding.py → openai_.py

@@ -21,9 +21,9 @@ def get_embedding(text):
         print(f"Error in get_embedding: {e}")
         return None
 
-def handle_user_query(query, db, collection):
+def handle_user_prompt(prompt, db, collection):
     # Assuming vector_search returns a list of dictionaries with keys 'title' and 'plot'
-    get_knowledge = vector_search(query, db, collection)
+    get_knowledge = vector_search(prompt, db, collection)
 
     # Check if there are any results
     if not get_knowledge:
@@ -47,7 +47,7 @@ def handle_user_query(query, db, collection):
                 "content": "You are a airbnb listing recommendation system."},
             {
                 "role": "user", 
-                "content": f"Answer this user query: {query} with the following context:\n{search_results_df}"
+                "content": f"Answer this user prompt: {prompt} with the following context:\n{search_results_df}"
             }
         ]
     )
@@ -55,7 +55,7 @@ def handle_user_query(query, db, collection):
     system_response = completion.choices[0].message.content
 
     # Print User Question, System Response, and Source Information
-    print(f"- User Question:\n{query}\n")
+    print(f"- User Question:\n{prompt}\n")
     print(f"- System Response:\n{system_response}\n")
 
     # Display the DataFrame as an HTML table