Mahbodez commited on
Commit
2f91a49
1 Parent(s): 72eff0b

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -60
app.py DELETED
@@ -1,60 +0,0 @@
1
- import gradio as gr
2
- import interface
3
- import utils
4
- import treegraph as tg
5
- import os
6
-
7
- user = os.environ.get("user")
8
- passwd = os.environ.get("pass")
9
-
10
- system_prompt = """
11
- You are a critical AI radiology assistant.
12
- You are helping a radiologist correctly fill out a radiology report.
13
- The report is regarding a Knee MRI.
14
- """
15
-
16
- graph, nodes_dict = tg.build_tree_from_file("knee_template.json")
17
- report_interface = interface.ReportChecklistInterface(
18
- llm=utils.LLM(model="gpt-3.5-turbo"),
19
- system_prompt=system_prompt,
20
- graph=graph,
21
- nodes_dict=nodes_dict,
22
- )
23
-
24
- if report_interface.prime_model() is False:
25
- print("Model priming failed. Please try again.")
26
- exit()
27
- else:
28
- print("Model priming successful.")
29
-
30
-
31
- def check_report(report):
32
- results = report_interface.process_input(report)
33
- if results == "quit":
34
- quit_fn()
35
- elif results == "help":
36
- return report_interface.help_message
37
- elif results == "exception":
38
- return "An exception occurred. Please try again."
39
- else:
40
- return results
41
-
42
-
43
- with gr.Blocks(theme="Taithrah/Minimal") as demo:
44
- gr.Markdown("## Radiology Report Assistant")
45
- gr.Markdown(report_interface.help_message)
46
-
47
- report_textbox = gr.TextArea(label="Report", lines=20, max_lines=50)
48
- check_btn = gr.Button(
49
- value="Check Report",
50
- )
51
- clear_btn = gr.ClearButton(
52
- value="Clear Messages",
53
- )
54
-
55
- results_textbox = gr.TextArea(label="Results", lines=20, max_lines=50)
56
- clear_btn.add([results_textbox, report_textbox])
57
-
58
- check_btn.click(fn=check_report, inputs=[report_textbox], outputs=[results_textbox])
59
-
60
- demo.launch(auth=(user, passwd))