Update app.py

app.py

@@ -150,7 +150,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     gr.Markdown("""
     # Arabic Diacritization & Reading Assessment
     1. Enter undiacritized Arabic text → Diacritize.
-    2. Read aloud & record/upload audio → Transcribe & Compare.
+    2. Optionally edit the diacritized result.
+    3. Record/upload audio → Transcribe & Compare.
     """)
     ref_state = gr.State("")
 
@@ -159,13 +160,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
             text_in = gr.Textbox(label="Undiacritized Arabic Text", lines=3, text_align="right")
             model_sel = gr.Dropdown(choices=["Encoder-Only","Encoder-Decoder"], value="Encoder-Only", label="Model")
             diac_btn = gr.Button("Diacritize Text")
-            diac_out = gr.Textbox(label="Diacritized Text (Reference)", lines=3, interactive=True, text_align="right")
-            diac_out.change(
-                fn=lambda text: text,  # Just pass the value through
-                inputs=diac_out,
-                outputs=reference_text_state
-            )
+            diac_out = gr.Textbox(label="Diacritized Text (Reference)", lines=3, text_align="right", interactive=True)
             diac_btn.click(fn=diacritize_text, inputs=[model_sel, text_in], outputs=[diac_out, ref_state])
+            diac_out.change(fn=lambda text: text, inputs=diac_out, outputs=ref_state)
 
         with gr.Column(scale=1):
             audio_in = gr.Audio(label="Record/Upload Audio", type="filepath")