Update written_module.py

written_module.py

@@ -60,19 +60,22 @@ def written_dashboard(nickname):
     with gr.Column() as written_panel:
         gr.Markdown("""
         <div id="header" style="text-align: center;">
+            <img src="file/images/chatter_owl.png" width="120">
             <h2>✍️ Written Communication with <strong>Chatter the Owl</strong></h2>
             <p>Paste or upload your writing. Select your goal and focus, and receive detailed feedback, improvement, and progress tracking.</p>
         </div>
         """)
 
-        with gr.Row(elem_classes=["top-control-row"]):
+        with gr.Row():
             language_dropdown = gr.Dropdown(label="🌍 Language", choices=list(LANG_CODES.keys()), value="English")
             goal_dropdown = gr.Dropdown(label="🎯 Writing Goal", choices=["Essay for school", "Job application", "Blog post", "Creative writing", "General improvement"], value="General improvement")
             focus_checkboxes = gr.CheckboxGroup(label="🧠 Focus Areas", choices=["Clarity", "Organization", "Grammar", "Vocabulary", "Tone", "Creativity"], value=["Clarity", "Organization", "Grammar", "Vocabulary", "Tone", "Creativity"])
 
-        with gr.Row(elem_classes=["audio-row"]):
+        with gr.Row():
             input_text = gr.Textbox(label="📝 Your Writing", lines=8, elem_id="input_text", elem_classes=["text_area"])
-            file_input = gr.File(label="📄 Upload Text File", file_types=[".txt", ".md"])
+            submit_button = gr.Button("🚀 Submit", variant="primary")
+
+        file_input = gr.File(label="📄 Upload Text File", file_types=[".txt", ".md"])
 
         feedback_box = gr.Textbox(label="💡 Feedback", interactive=False, elem_id="feedback_box")
         improved_box = gr.Textbox(label="🎯 Improved Version", interactive=False, elem_id="improved_box")
@@ -115,7 +118,7 @@ def written_dashboard(nickname):
 
             return feedback, improved, session_table, score_plot, trend_plot, gr.update(visible=bool(milestone_msg), value=milestone_msg)
 
-        input_text.change(
+        submit_button.click(
             fn=tutor_written,
             inputs=[input_text, language_dropdown, goal_dropdown, focus_checkboxes],
             outputs=[feedback_box, improved_box, history_table, score_chart, trend_chart, milestone_box]