Update src/txagent/txagent.py

src/txagent/txagent.py

@@ -761,17 +761,17 @@ Generate **one summarized sentence** about "function calls' responses" with nece
         return updated_attributes
 
     def run_gradio_chat(self, message: str,
-                        history: list,
-                        temperature: float,
-                        max_new_tokens: int,
-                        max_token: int,
-                        call_agent: bool,
-                        conversation: gr.State,
-                        max_round: int = 20,
-                        seed: int = None,
-                        call_agent_level: int = 0,
-                        sub_agent_task: str = None,
-                        uploaded_files: list = None) -> str:
+                            history: list,
+                            temperature: float,
+                            max_new_tokens: int,
+                            max_token: int,
+                            call_agent: bool,
+                            conversation: gr.State,
+                            max_round: int = 20,
+                            seed: int = None,
+                            call_agent_level: int = 0,
+                            sub_agent_task: str = None,
+                            uploaded_files: list = None) -> str:
         """
         Generate a streaming response using the llama3-8b model.
         Args:
@@ -784,29 +784,9 @@ Generate **one summarized sentence** about "function calls' responses" with nece
         """
         print("\033[1;32;40mstart\033[0m")
         print("len(message)", len(message))
-
-        if 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 = []
@@ -845,7 +825,6 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                         temperature=temperature)
                     history.extend(current_gradio_history)
                     if special_tool_call == 'Finish':
-                        yield history
                         next_round = False
                         conversation.extend(function_call_messages)
                         return function_call_messages[0]['content']
@@ -853,7 +832,6 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                         history.append(
                             ChatMessage(role="assistant", content=history[-1].content))
                         yield history
-                        next_round = False
                         return history[-1].content
                     if (self.enable_summary or token_overflow) and not call_agent:
                         if token_overflow:
@@ -864,9 +842,8 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                         enable_summary=enable_summary)
                     if function_call_messages is not None:
                         conversation.extend(function_call_messages)
-                        formated_md_function_call_messages = tool_result_format(
-                            function_call_messages)
-                        yield history
+                        # Hide intermediate output
+                        pass
                     else:
                         next_round = False
                         conversation.extend(
@@ -893,18 +870,12 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                     if each.metadata is not None:
                         each.metadata['status'] = 'done'
                 if '[FinalAnswer]' in last_thought:
-                    final_thought, final_answer = last_thought.split(
-                        '[FinalAnswer]')
+                    _, final_answer = last_thought.split('[FinalAnswer]', 1)
                     history.append(
-                        ChatMessage(role="assistant",
-                                    content=final_thought.strip())
-                    )
-                    yield history
-                    history.append(
-                        ChatMessage(
-                            role="assistant", content="**Answer**:\n"+final_answer.strip())
+                        ChatMessage(role="assistant", content=final_answer.strip())
                     )
                     yield history
+                    return
                 else:
                     history.append(ChatMessage(
                         role="assistant", content=last_thought))
@@ -920,16 +891,9 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                         if each.metadata is not None:
                             each.metadata['status'] = 'done'
                     if '[FinalAnswer]' in last_thought:
-                        final_thought, final_answer = last_thought.split(
-                            '[FinalAnswer]')
-                        history.append(
-                            ChatMessage(role="assistant",
-                                        content=final_thought.strip())
-                        )
-                        yield history
+                        _, final_answer = last_thought.split('[FinalAnswer]', 1)
                         history.append(
-                            ChatMessage(
-                                role="assistant", content="**Answer**:\n"+final_answer.strip())
+                            ChatMessage(role="assistant", content=final_answer.strip())
                         )
                         yield history
                 else:
@@ -948,19 +912,12 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                         each.metadata['status'] = 'done'
                 if '[FinalAnswer]' in last_thought or '"name": "Finish",' in last_outputs_str:
                     if '[FinalAnswer]' in last_thought:
-                        final_thought, final_answer = last_thought.split('[FinalAnswer]', 1)
+                        _, final_answer = last_thought.split('[FinalAnswer]', 1)
                     else:
-                        final_thought = ""
                         final_answer = last_thought
                     history.append(
-                        ChatMessage(role="assistant",
-                                    content=final_thought.strip())
-                    )
-                    yield history
-                    history.append(
-                        ChatMessage(
-                            role="assistant", content="**Answer**:\n" + final_answer.strip())
+                        ChatMessage(role="assistant", content=final_answer.strip())
                     )
                     yield history
             else:
-                return None
+                return None