Update app.py

app.py

@@ -8,30 +8,10 @@ RUNPOD_ENDPOINT = os.environ.get("RUNPOD_ENDPOINT")
 auth = [(os.environ.get("S2S_USER"), os.environ.get("S2S_PW")), (os.environ.get("POSOS_USER"), os.environ.get("POSOS_PW"))]
 
 def update_message(request: gr.Request):
-    m = gr.Markdown(f"Welcome, {request.username}")
     debug = False
     if request.username == os.environ.get("S2S_USER"):
         debug = True
-    with gr.Tab("Speech2text prescription"):
-        with gr.Row():
-            with gr.Column():
-                audio_file = gr.Audio(sources=["upload", "microphone"], type="numpy", label="Audio")
-                submit_btn = gr.Button("Submit", variant="primary")
-            with gr.Column():
-                if debug:
-                    debug_transcript = gr.Textbox(label="Transcription Debug")
-                    transcript = gr.Textbox(label="Transcription")
-                else:
-                    transcript = gr.Textbox(label="Transcription")
-        submit_btn.click(process_transcribe, inputs=[audio_file], outputs=[transcript])
-    return 
-
-def debug(request):
-    debug = False
-    print(request.username)
-    if request.username == os.environ.get("S2S_USER"):
-        debug = True
-    return debug
+    return f"Welcome, {request.username}", debug
 
 def process_transcribe(file):
     print(file)
@@ -49,6 +29,21 @@ def process_transcribe(file):
     return raw_text
 
 with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as iface:
-    iface.load(update_message, None)
+    m = gr.Markdown()
+    debug = False
+    iface.load(update_message, None, [m,debug])
+    
+    with gr.Tab("Speech2text prescription"):
+        with gr.Row():
+            with gr.Column():
+                audio_file = gr.Audio(sources=["upload", "microphone"], type="numpy", label="Audio")
+                submit_btn = gr.Button("Submit", variant="primary")
+            with gr.Column():
+                if debug:
+                    debug_transcript = gr.Textbox(label="Transcription Debug")
+                    transcript = gr.Textbox(label="Transcription")
+                else:
+                    transcript = gr.Textbox(label="Transcription")
+        submit_btn.click(process_transcribe, inputs=[audio_file], outputs=[transcript])
     logout_button = gr.Button("Logout", link="/logout")
 iface.launch(auth=auth)