Update app.py

app.py

@@ -55,7 +55,7 @@ def circular_html(ai_percent):
         </div>
     </div>
     """
-    
+
 # ===== Chunking for Large Input Support =====
 def chunk_text(text, max_tokens=300):
     paragraphs = text.split("\n\n")
@@ -97,6 +97,19 @@ def paraphrase(text):
         for chunk in chunks
     )
 
+# ===== Apply Enhancements Based on Checkboxes =====
+def apply_selected_enhancements(text, fix, tone, fluency, para):
+    result = text
+    if fix:
+        result = fix_grammar(result)
+    if tone:
+        result = improve_tone(result)
+    if fluency:
+        result = improve_fluency(result)
+    if para:
+        result = paraphrase(result)
+    return result
+
 # ===== AI Detection and Visualization =====
 def detect_ai_percent(text):
     inputs = detector_tokenizer(text, return_tensors="pt", truncation=True, padding=True)
@@ -134,17 +147,20 @@ with gr.Blocks() as demo:
         input_text = gr.Textbox(lines=12, label="Input Text")
     load_btn.click(fn=load_file, inputs=file_input, outputs=input_text)
 
+    gr.Markdown("## ✅ Select Enhancements to Apply")
     with gr.Row():
-        btn_grammar = gr.Button("✔️ Fix Grammar")
-        btn_tone = gr.Button("🎯 Improve Tone")
-        btn_fluency = gr.Button("🔄 Improve Fluency")
-        btn_paraphrase = gr.Button("🌀 Paraphrase")
+        check_fix = gr.Checkbox(label="✔️ Fix Grammar")
+        check_tone = gr.Checkbox(label="🎯 Improve Tone")
+        check_fluency = gr.Checkbox(label="🔄 Improve Fluency")
+        check_paraphrase = gr.Checkbox(label="🌀 Paraphrase")
 
+    run_selected = gr.Button("🚀 Run Selected Enhancements")
     output_text = gr.Textbox(lines=12, label="Output")
-    btn_grammar.click(fn=fix_grammar, inputs=input_text, outputs=output_text)
-    btn_tone.click(fn=improve_tone, inputs=input_text, outputs=output_text)
-    btn_fluency.click(fn=improve_fluency, inputs=input_text, outputs=output_text)
-    btn_paraphrase.click(fn=paraphrase, inputs=input_text, outputs=output_text)
+    run_selected.click(
+        fn=apply_selected_enhancements,
+        inputs=[input_text, check_fix, check_tone, check_fluency, check_paraphrase],
+        outputs=output_text
+    )
 
     gr.Markdown("## 🕵️ AI Detection")
     detect_btn = gr.Button("Detect AI Probability")