Update app.py

app.py

@@ -8,6 +8,13 @@ import gradio
 from gradio_client import Client
 import autogen
 import panel as pn
+import os
+from unstructured.ingest.interfaces import PartitionConfig, ProcessorConfig, ReadConfig  
+from unstructured.ingest.runner import LocalRunner  
+from unstructured.partition.auto import partition  
+from langchain.document_loaders import UnstructuredFileLoader  
+
+
 from autogen_utils import (
     MathUserProxyAgent,
     RetrieveUserProxyAgent,
@@ -21,6 +28,29 @@ from custom_widgets import RowAgentWidget
 from panel.chat import ChatInterface
 from panel.widgets import Button, CodeEditor, PasswordInput, Switch, TextInput
 
+def process_file_with_unstructured(file_path):
+    # Set up the LocalRunner
+    runner = LocalRunner(
+        processor_config=ProcessorConfig(
+            verbose=True,
+            output_dir="local-ingest-output",
+            num_processes=2,
+        ),
+        read_config=ReadConfig(),
+        partition_config=PartitionConfig(
+            partition_by_api=True,
+            api_key=os.getenv("UNSTRUCTURED_API_KEY"),
+        ),
+    )
+    runner.run(input_path=file_path, recursive=True)
+    elements = partition(filename=file_path, content_type="application/pdf")  
+    loader = UnstructuredFileLoader(file_path)
+    docs = loader.load()
+    raw_text = docs[0].page_content
+
+    return raw_text
+
+
 pn.extension("codeeditor")
 template = pn.template.BootstrapTemplate(title=TITLE)
 
@@ -316,7 +346,18 @@ def init_groupchat(event, collection_name):
     return agents, manager, groupchat
 
 
-async def agents_chat(init_sender, manager, contents, agents):
+async def agents_chat(init_sender, manager, contents, agents, RAG):
+    # Check if a file is uploaded
+    if RAG.value:
+        # Save the file and process it
+        file_path = "path/to/saved/file"  # Define the path to save the uploaded file
+        RAG.save(file_path)
+        raw_text = process_file_with_unstructured(file_path)
+
+        # Prepend the extracted text to the contents
+        contents = raw_text + ' ' + contents
+
+    # Determine the recipient
     recipient = (
         manager
         if len(agents) > 2
@@ -324,6 +365,8 @@ async def agents_chat(init_sender, manager, contents, agents):
         if agents[1] != init_sender
         else agents[0]
     )
+
+    # Initiate chat
     if isinstance(init_sender, (RetrieveUserProxyAgent, MathUserProxyAgent)):
         await init_sender.a_initiate_chat(recipient, problem=contents)
     else: