comparison option

app.py

@@ -13,7 +13,8 @@ def indexing(llm,path):
     query_engine = index.as_query_engine()
     cl.user_session.set("engine", query_engine)
 
-def qa(sp,engine,message):
+def qa(sp,message):
+    engine=cl.user_session.get("engine")
     message=message.content
     ques=sp+" "+message
     response=engine.query(ques)
@@ -47,13 +48,14 @@ async def factory():
 
     else:
         await cl.Message(author="Beast",content="Okay, then you can start asking your questions!!").send()
+
         
 @cl.on_message
 async def main(message: cl.Message):
     service_client=cl.user_session.get("service_client")
     engine = cl.user_session.get("engine")
     llm=cl.user_session.get("llm")
-    sp=cl.user_session.get("sp")
+    sp=cl.user_session.get("sp")    
     
     if sp==None:
         sp=""
@@ -70,25 +72,39 @@ async def main(message: cl.Message):
         if go:
             msg = cl.Message(author="Beast",content=f"Processing `{pdf.name}`...")
             await msg.send()    
-            query_engine = await cl.make_async(indexing)(llm,pdf.path)
+            await cl.make_async(indexing)(llm,pdf.path)
             msg.content = f"`{pdf.name}` processed."
             await msg.update()
             msg = cl.Message(author="Beast",content=f"Generating Response...")
             await msg.send()
-            response =await cl.make_async(qa)(sp,query_engine,message)
+            response =await cl.make_async(qa)(sp,message)
             print(response)
             msg.content = str(response)
             await msg.update()
+            comp = await cl.AskActionMessage(author="Beast",
+            content="Do you want answer without RAG?",
+            actions=[
+                cl.Action(name="yes", value="yes", label="✅ Yes"),
+                cl.Action(name="no", value="no", label="❌ No"),
+            ],
+        ).send()
     
     elif not message.elements and engine!=None:
         msg = cl.Message(author="Beast",content=f"Generating Response...")
         await msg.send()
-        response =await cl.make_async(qa)(sp,engine,message)
+        response =await cl.make_async(qa)(sp,message)
         print(response)
         msg.content = str(response)
         await msg.update()
+        comp = await cl.AskActionMessage(author="Beast",
+            content="Do you want answer without RAG?",
+            actions=[
+                cl.Action(name="yes", value="yes", label="✅ Yes"),
+                cl.Action(name="no", value="no", label="❌ No"),
+            ],
+        ).send()
     
-    elif not message.elements and engine==None:
+    if (not message.elements and engine==None) or (comp.get("value") == "yes"):
         msg = cl.Message(author="Beast",content=f"Generating Response...")
         await msg.send()
         payload = {
@@ -103,8 +119,4 @@ async def main(message: cl.Message):
         await msg.update()
     
     else:
-        cl.Message(author="Beast",content="Broken ;(")
-        
-    
-    
-   
+        cl.Message(author="Beast",content="Broken ;(")