RAG1

app.py

@@ -4,6 +4,7 @@ import os
 import base64
 import time
 import copy
+from rag import get_top_k
 from dotenv import load_dotenv
 # Load environment variables from .env file
 load_dotenv()
@@ -22,6 +23,13 @@ with gr.Blocks() as demo:
     )
     clear = gr.ClearButton([chat_input, chatbot])
 
+    def use_rag(query, k=3):
+        """
+        Use RAG to get top k results for the query.
+        """
+        results = get_top_k(query, k=k)
+        return results
+    
     def encode_image(image_path):
         with open(image_path, "rb") as f:
             return base64.b64encode(f.read()).decode("utf-8")