Ali2206 commited on
Commit
8cba180
·
verified ·
1 Parent(s): 7e2a12c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -137,20 +137,23 @@ def create_agent():
137
  logger.error(f"Failed to create agent: {str(e)}")
138
  raise
139
 
140
- def respond(msg, chat_history, temperature, max_new_tokens, max_tokens, multi_agent, conversation, max_round):
141
- if not msg or len(msg.strip()) <= 10:
142
- return chat_history + [["assistant", "Please provide a valid message with a string longer than 10 characters."]]
 
 
 
143
 
144
- # Append user message in tuple format
145
- chat_history = chat_history + [["user", msg]]
146
 
147
  print("\n==== DEBUG ====")
148
- print("User Message:", msg)
149
  print("Chat History:", chat_history)
150
  print("================\n")
151
 
152
  try:
153
- # Convert to TxAgent format
154
  formatted_history = [(role, content) for role, content in chat_history]
155
 
156
  response_generator = agent.run_gradio_chat(
 
137
  logger.error(f"Failed to create agent: {str(e)}")
138
  raise
139
 
140
+ def respond(message_input, chat_history, temperature, max_new_tokens, max_tokens, multi_agent, conversation, max_round):
141
+ # Ensure the message is a proper string
142
+ message = message_input.strip() if isinstance(message_input, str) else ""
143
+
144
+ if len(message) <= 10:
145
+ return chat_history + [["assistant", "Hi, I am TxAgent, an assistant for answering biomedical questions. Please provide a valid message with a string longer than 10 characters."]]
146
 
147
+ # Add user message to history
148
+ chat_history = chat_history + [["user", message]]
149
 
150
  print("\n==== DEBUG ====")
151
+ print("User Message:", message)
152
  print("Chat History:", chat_history)
153
  print("================\n")
154
 
155
  try:
156
+ # Format for model: list of (role, message) tuples
157
  formatted_history = [(role, content) for role, content in chat_history]
158
 
159
  response_generator = agent.run_gradio_chat(