Update src/txagent/txagent.py

src/txagent/txagent.py

@@ -785,26 +785,28 @@ 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 = []
+            file_content_blocks = []
+            for file in uploaded_files:
+                if file.name.endswith(".pdf"):
+                    try:
+                        import fitz  # PyMuPDF
+                        doc = fitz.open(file.name)
+                        text = ""
+                        for page in doc:
+                            text += page.get_text()
+                        if text.strip():
+                            file_content_blocks.append(f"[FILE CONTENT]\n{text.strip()}\n")
+                    except Exception as e:
+                        print(f"Error reading PDF: {e}")
+
+            if file_content_blocks:
+                message = "\n".join(file_content_blocks) + "\n[USER PROMPT]\n" + message
 
         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."
+
         outputs = []
         outputs_str = ''
         last_outputs = []