attatchment enabled

app.py

@@ -2,7 +2,9 @@ import chainlit as cl
 from llama_index.llms import MonsterLLM
 from llama_index import VectorStoreIndex,SimpleDirectoryReader, ServiceContext
 
-def indexing(llm,path):
+def indexing(llm,path=None):
+    if path==None:
+        path="data.txt"
     documents = SimpleDirectoryReader(input_files=[path]).load_data()
     print("loading done")
     service_context = ServiceContext.from_defaults(
@@ -31,22 +33,18 @@ async def factory():
     print(auth)
     model = 'deploy-llm'
     llm = MonsterLLM(model=model,base_url=url['output'],monster_api_key=auth['output'],temperature=0.75, context_window=1024)
-    files = None
-    while files is None:
-        files = await cl.AskFileMessage(author="Beast",
-            content="Please upload a PDF file to begin!",
-            accept=["application/pdf"],
-            max_size_mb=20,
-            timeout=180,
-        ).send()
+    cl.user_session.set("llm", llm)
+    # files = None
+    # while files is None:
+    #     files = await cl.AskFileMessage(author="Beast",
+    #         content="Please upload a PDF file to begin!",
+    #         accept=["application/pdf"],
+    #         max_size_mb=20,
+    #         timeout=180,
+    #     ).send()
 
-    pdf = files[0]
-    print(pdf)
-    msg = cl.Message(author="Beast",content=f"Processing `{pdf.name}`...")
-    await msg.send()    
-    query_engine = await cl.make_async(indexing)(llm,pdf.path)
-    msg.content = f"`{pdf.name}` processed."
-    await msg.update()
+    # pdf = files[0]
+    # print(pdf)
     res = await cl.AskActionMessage(author="Beast",
         content="Do you want to enter system prompt?",
         actions=[
@@ -54,7 +52,7 @@ async def factory():
             cl.Action(name="no", value="no", label="❌ No"),
         ],
     ).send()
-
+    query_engine = await cl.make_async(indexing)(llm)
     if res and res.get("value") == "yes":
         sp = await cl.AskUserMessage(author="Beast",content="Enter system prompt").send()
         await cl.Message(author="Beast",content="Noted. Go ahead as your questions!!").send()
@@ -66,14 +64,39 @@ async def factory():
     
 @cl.on_message
 async def main(message: cl.Message):
-    msg = cl.Message(author="Beast",content=f"Processing...", disable_feedback=False)
-    await msg.send()
     engine = cl.user_session.get("engine")
+    llm=cl.user_session.get("llm")
     sp=cl.user_session.get("sp")
     if sp==None:
         sp=""
-    response =await cl.make_async(qa)(sp,engine,message)
-    print(response)
-    msg.content = str(response)
-    await msg.update()
+    if not message.elements:
+        msg = cl.Message(author="Beast",content=f"Processing...", disable_feedback=False)
+        await msg.send()
+        response =await cl.make_async(qa)(sp,engine,message)
+        print(response)
+        msg.content = str(response)
+        await msg.update()
+    elif message.elements:
+        go=True
+        for file in message.elements: 
+            if "pdf" in file.mime:
+                pdf=file
+            else:
+                await cl.Message(author="Beast",content="We only support PDF for now").send()
+                go=False
+                break
+        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)
+            msg.content = f"`{pdf.name}` processed."
+            await msg.update()
+            response =await cl.make_async(qa)(sp,query_engine,message)
+            print(response)
+            msg.content = str(response)
+            msg.author="Beast"
+            await msg.send()
+    
+    
+    
    

requirements.txt

@@ -1,4 +1,5 @@
 monsterapi
 chainlit
 pypdf
-git+https://github.com/Vikasqblocks/llama_index.git@f2f04654e9f2cbf1bf765b0d575a6af1f899b18e
+git+https://github.com/Vikasqblocks/llama_index.git@f2f04654e9f2cbf1bf765b0d575a6af1f899b18e
+sentence-transformers