Update app.py

app.py

@@ -16,7 +16,11 @@ RAG_ADVANCED = "Advanced RAG"
 logging.basicConfig(stream = sys.stdout, level = logging.INFO)
 logging.getLogger().addHandler(logging.StreamHandler(stream = sys.stdout))
     
-def invoke(openai_api_key, prompt, rag_option):
+def invoke(openai_api_key, 
+           prompt, 
+           accomodates, 
+           bedrooms, 
+           rag_option):
     if not openai_api_key:
         raise gr.Error("OpenAI API Key is required.")
     if not prompt:
@@ -30,8 +34,15 @@ def invoke(openai_api_key, prompt, rag_option):
         if (RAG_INGESTION):
             return rag_ingestion(collection)
         else:
-            retrieval_result = rag_retrieval(openai_api_key, prompt, db, collection)
-            inference_result = rag_inference(openai_api_key, prompt, retrieval_result)
+            retrieval_result = rag_retrieval(openai_api_key, 
+                                             prompt, 
+                                             accomodates,
+                                             bedrooms,
+                                             db, 
+                                             collection)
+            inference_result = rag_inference(openai_api_key, 
+                                             prompt, 
+                                             retrieval_result)
             print("###")
             print(inference_result)
             print("###")
@@ -45,6 +56,8 @@ demo = gr.Interface(
     fn = invoke, 
     inputs = [gr.Textbox(label = "OpenAI API Key", type = "password", lines = 1), 
               gr.Textbox(label = "Prompt", value = PROMPT, lines = 1),
+              gr.Number(label = "Accomodates", value = 2),
+              gr.Number(label = "Bedrooms", value = 1),
               gr.Radio([RAG_OFF, RAG_NAIVE, RAG_ADVANCED], label = "Retrieval-Augmented Generation", value = RAG_ADVANCED)],
     outputs = [gr.Markdown(label = "Completion", value = os.environ["OUTPUT"], line_breaks = True, sanitize_html = False)],
     title = "Context-Aware Reasoning Application",