Update app.py

app.py

@@ -27,11 +27,34 @@ if report_interface.prime_model() is False:
 else:
     print("Model priming successful.")
 
-with gr.Blocks(theme="finlaymacklon/smooth_slate") as demo:
+running = False
+
+
+def check_report(report):
+    if running:
+        results = report_interface.process_input(report)
+        if results == "quit":
+            quit_fn()
+        elif results == "help":
+            return report_interface.help_message
+        elif results == "exception":
+            return "An exception occurred. Please try again."
+        else:
+            return results
+    else:
+        return "Model has been stopped."
+
+
+def quit_fn():
+    global running
+    running = False
+    return "Model has been stopped."
+
+
+with gr.Blocks(theme="soft") as demo:
     gr.Markdown("## Radiology Report Assistant")
     gr.Markdown(report_interface.help_message)
 
-    running = gr.components.Variable(True)
     report_textbox = gr.TextArea(label="Report", lines=20, max_lines=50)
     check_btn = gr.Button(
         value="Check Report",
@@ -45,24 +68,7 @@ with gr.Blocks(theme="finlaymacklon/smooth_slate") as demo:
     results_textbox = gr.TextArea(label="Results", lines=20, max_lines=50)
     clear_btn.add([results_textbox, report_textbox])
 
-    def check_report(report):
-        if running.value is True:
-            results = report_interface.process_input(report)
-            if results == "quit":
-                quit_fn()
-            elif results == "help":
-                return report_interface.help_message
-            elif results == "exception":
-                return "An exception occurred. Please try again."
-            else:
-                return results
-        else:
-            return "Model has been stopped."
-
-    def quit_fn():
-        return "Model has been stopped.", False
-
     check_btn.click(fn=check_report, inputs=[report_textbox], outputs=[results_textbox])
-    quit_btn.click(fn=quit_fn, outputs=[results_textbox, running])
+    quit_btn.click(fn=quit_fn, outputs=[results_textbox])
 
 demo.launch(auth=(user, passwd))