Update app.py

app.py

@@ -187,7 +187,7 @@ def query_chroma(vectorstore, query, k):
     return chunks_with_references
 
 def rag_workflow(query):
-    docs = query_chroma(vectorstore, query, k=5)
+    docs = query_chroma(vectorstore, query, k=20)
     context = "\n\n".join([doc for doc, _ in docs])
     references = "\n".join([f"[{i+1}] {ref}" for i, (_, ref) in enumerate(docs)])
     print(f"Context for the query:\n{context}\n")
@@ -221,6 +221,11 @@ def add_text(history, text):
     history = history + [(text, None)]
     yield history, ""
 
+
+
+import gradio as gr
+
+
 def bot_kadi(history):
     user_query = history[-1][0]
     response, references = rag_workflow(user_query)