Update app.py

app.py

@@ -99,26 +99,29 @@ def handle_magic_click(current_text, current_button_text):
     result = magic_function(current_text)
     category_fix = result.get("category_fix", "")
     suggestion = result.get("suggestion", "")
+
+    print(current_text)
+    print(category_fix)
+    print(suggestion)
+    print()
     
     if category_fix == "COMPLIANT":
         # COMPLIANT case - no changes to input, green background, lock, add checklist, change button to Reset
         return (
-            gr.update(value=current_text, elem_classes="success-text", interactive=False),  # text_area - keep original, green, lock
+            gr.update(label="Your Text ✓", "value=current_text, elem_classes="success-text", interactive=False),  # text_area - keep original, green, lock
             current_text,  # previous_text_area - store current
             gr.update(visible=False),  # previous_text_area visibility - hide
             gr.update(value=f"**{category_fix}**", visible=True),  # category_label - show category in bold
             gr.update(value="Reset"),  # button text - change to Reset
-            "Your Text ✓"  # text_area label - add checklist symbol
         )
     else:
         # Non-compliant case - show suggestion, show previous text, show category
         return (
-            gr.update(value=suggestion, elem_classes="", interactive=True),  # text_area - show suggestion, normal style, unlocked
+            gr.update(label="Your Text", value=suggestion, elem_classes="", interactive=True),  # text_area - show suggestion, normal style, unlocked
             current_text,  # previous_text_area - store current
             gr.update(visible=True),  # previous_text_area visibility - show
             gr.update(value=f"**{category_fix}**", visible=True),  # category_label - show category in bold
             gr.update(value="Magic Button"),  # button text - keep as Magic Button
-            "Your Text"  # text_area label - normal
         )
 
 
@@ -135,7 +138,7 @@ with gr.Blocks(css=".success-text { background-color: #d4edda !important; }") as
     magic_button.click(
         fn=handle_magic_click,
         inputs=[text_area, magic_button],
-        outputs=[text_area, previous_text_area, previous_text_area, category_label, magic_button, text_area]
+        outputs=[text_area, previous_text_area, previous_text_area, category_label, magic_button]
     )
 
 if __name__ == "__main__":