Update src/txagent/txagent.py

src/txagent/txagent.py

@@ -770,7 +770,8 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                         max_round: int = 20,
                         seed: int = None,
                         call_agent_level: int = 0,
-                        sub_agent_task: str = None) -> str:
+                        sub_agent_task: str = None,
+                        uploaded_files: list = None) -> str:
         """
         Generate a streaming response using the llama3-8b model.
         Args:
@@ -783,6 +784,24 @@ Generate **one summarized sentence** about "function calls' responses" with nece
         """
         print("\033[1;32;40mstart\033[0m")
         print("len(message)", len(message))
+
+        # Handle uploaded files
+        if uploaded_files:
+            self.uploaded_files = uploaded_files
+            print("\U0001F4C1 Uploaded files received:")
+            file_context = ""
+            for f in uploaded_files:
+                print(" -", f.name if hasattr(f, 'name') else f)
+                try:
+                    with open(f.name, 'r', encoding='utf-8', errors='ignore') as fobj:
+                        content = fobj.read()
+                        file_context += f"\n\n[File: {f.name}]\n{content}\n"
+                except Exception as e:
+                    file_context += f"\n\n[File: {f.name}]\n(Error reading file: {e})\n"
+            message = f"The user uploaded the following files to help you answer:\n{file_context}\n\n{message}"
+        else:
+            self.uploaded_files = []
+
         if len(message) <= 10:
             yield "Hi, I am TxAgent, an assistant for answering biomedical questions. Please provide a valid message with a string longer than 10 characters."
             return "Please provide a valid message."
@@ -942,4 +961,4 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                     )
                     yield history
             else:
-                return None
+                return None