Update app.py

app.py

@@ -29,13 +29,15 @@ def invoke(openai_api_key, prompt, rag_option):
         os.environ["OPENAI_API_KEY"] = openai_api_key
 
         ###
-        query = """
+        prompt = """
         I want to stay in a place that's warm and friendly, 
         and not too far from resturants, can you recommend a place? 
         Include a reason as to why you've chosen your selection.
         """
-        handle_user_query(query, db, collection)
+        result = handle_user_prompt(prompt, db, collection)
         ###
+
+        del os.environ["OPENAI_API_KEY"]
         
         """
         if (RAG_INGESTION):
@@ -65,7 +67,7 @@ def invoke(openai_api_key, prompt, rag_option):
             del os.environ["OPENAI_API_KEY"]
         """
 
-        return "TODO"
+        return result
 
 gr.close_all()