Ali2206 commited on
Commit
2bea759
·
verified ·
1 Parent(s): 924aa8a

Update ui/ui_core.py

Browse files
Files changed (1) hide show
  1. ui/ui_core.py +2 -5
ui/ui_core.py CHANGED
@@ -8,7 +8,6 @@ import gradio as gr
8
  sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
9
  from txagent.txagent import TxAgent
10
 
11
-
12
  def extract_all_text_from_csv_or_excel(file_path, progress=None, index=0, total=1):
13
  try:
14
  if file_path.endswith(".csv"):
@@ -23,7 +22,6 @@ def extract_all_text_from_csv_or_excel(file_path, progress=None, index=0, total=
23
  except Exception as e:
24
  return f"Error parsing file: {e}"
25
 
26
-
27
  def extract_all_text_from_pdf(file_path, progress=None, index=0, total=1):
28
  extracted = []
29
  try:
@@ -41,10 +39,9 @@ def extract_all_text_from_pdf(file_path, progress=None, index=0, total=1):
41
  except Exception as e:
42
  return f"Error parsing PDF: {e}"
43
 
44
-
45
  def create_ui(agent: TxAgent):
46
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
47
- gr.Markdown("<h1 style='text-align: center;'>📋 CPS: Clinical Patient Support System</h1>")
48
  chatbot = gr.Chatbot(label="CPS Assistant", height=600, type="messages")
49
 
50
  file_upload = gr.File(
@@ -107,4 +104,4 @@ def create_ui(agent: TxAgent):
107
  ["Is there anything abnormal in the attached blood work report?"]
108
  ], inputs=message_input)
109
 
110
- return demo
 
8
  sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
9
  from txagent.txagent import TxAgent
10
 
 
11
  def extract_all_text_from_csv_or_excel(file_path, progress=None, index=0, total=1):
12
  try:
13
  if file_path.endswith(".csv"):
 
22
  except Exception as e:
23
  return f"Error parsing file: {e}"
24
 
 
25
  def extract_all_text_from_pdf(file_path, progress=None, index=0, total=1):
26
  extracted = []
27
  try:
 
39
  except Exception as e:
40
  return f"Error parsing PDF: {e}"
41
 
 
42
  def create_ui(agent: TxAgent):
43
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
44
+ gr.Markdown("<h1 style='text-align: center;'>\ud83d\udccb CPS: Clinical Patient Support System</h1>")
45
  chatbot = gr.Chatbot(label="CPS Assistant", height=600, type="messages")
46
 
47
  file_upload = gr.File(
 
104
  ["Is there anything abnormal in the attached blood work report?"]
105
  ], inputs=message_input)
106
 
107
+ return demo