Update app.py

app.py

@@ -27,7 +27,6 @@ def extract_text_from_url(url):
     else:
         return "❌ Blog Error: Could not fetch content from the URL."
 
-
 # Review line-by-line
 def review_lines(text):
     lines = text.strip().split('\n')
@@ -35,19 +34,22 @@ def review_lines(text):
     for line in lines:
         if line.strip() == "":
             continue
-        prompt = f"Rewrite this to fix grammar, tone, and remove any offensive language:\n\n{line}"
+        prompt = (
+            f"Review this line for grammar, tone, and offensive language. "
+            f"Show suggested corrections with changes **highlighted** in Markdown (e.g., ~~wrong~~ → **right**):\n\n{line}"
+        )
         suggestion = reviewer(prompt, max_new_tokens=100)[0]['generated_text']
         suggestions.append((line, suggestion.strip()))
     return suggestions
 
-# Finalize by combining accepted lines
+# Finalize text
 def finalize_text(originals, suggestions, decisions):
     final = []
     for orig, sugg, keep in zip(originals, suggestions, decisions):
         final.append(sugg if keep else orig)
     return "\n".join(final)
 
-# Gradio UI
+# Gradio app
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as app:
     gr.Markdown("## ✨ BlogChecker AI\nSmart AI reviewer for blog content, with interactive approval and OCR image support.")
 
@@ -62,20 +64,8 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as app:
         review_btn = gr.Button("🧠 Review Content")
         finalize_btn = gr.Button("✅ Finalize Clean Blog")
 
-    line_outputs = []
-    decisions = []
-    originals = []
-    suggestions = []
-
-    with gr.Column() as dynamic_review_section:
-        for i in range(5):  # support 5 lines for demo; can be dynamic later
-            orig = gr.Textbox(label=f"Original Line {i+1}", interactive=False)
-            sugg = gr.Textbox(label=f"Suggested Line {i+1}", interactive=False)
-            accept = gr.Checkbox(label="✅ Accept Suggestion")
-            originals.append(orig)
-            suggestions.append(sugg)
-            decisions.append(accept)
-            line_outputs.append((orig, sugg, accept))
+    review_section = gr.Column(visible=False)
+    review_boxes = []  # Will store tuples: (orig_box, sugg_box, accept_checkbox)
 
     final_output = gr.Textbox(label="📦 Final Clean Blog", lines=10)
 
@@ -86,25 +76,26 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as app:
 
     def process_review(text):
         results = review_lines(text)
-        outputs = []
-        for i in range(len(line_outputs)):
-            if i < len(results):
-                outputs.extend([results[i][0], results[i][1], False])  # orig, suggestion, unselected
-            else:
-                outputs.extend(["", "", False])  # clear unused slots
-        return outputs
-
-    def collect_decisions(*args):
-        count = len(args) // 3
-        origs = args[:count]
-        suggs = args[count:2*count]
-        accepts = args[2*count:]
-        return finalize_text(origs, suggs, accepts)
-
+        review_section.children.clear()
+        review_boxes.clear()
+
+        for i, (orig, sugg) in enumerate(results):
+            with review_section:
+                orig_box = gr.Textbox(value=orig, label=f"Original Line {i+1}", interactive=False)
+                sugg_box = gr.Markdown(value=sugg, label=f"Suggested Edit {i+1}")
+                accept_box = gr.Checkbox(label="✅ Accept Suggestion", value=False)
+                review_boxes.append((orig_box, sugg_box, accept_box))
+        return gr.update(visible=True)
+
+    def collect_dynamic_decisions():
+        originals_vals = [box[0].value for box in review_boxes]
+        suggestions_vals = [box[1].value for box in review_boxes]
+        accepts_vals = [box[2].value for box in review_boxes]
+        return finalize_text(originals_vals, suggestions_vals, accepts_vals)
 
     # Wire actions
     extract_btn.click(fn=extract_both, inputs=[blog_url, image_url], outputs=combined_text)
-    review_btn.click(fn=process_review, inputs=combined_text, outputs=[el for group in line_outputs for el in group])
-    finalize_btn.click(fn=collect_decisions, inputs=originals + suggestions + decisions, outputs=final_output)
+    review_btn.click(fn=process_review, inputs=combined_text, outputs=review_section)
+    finalize_btn.click(fn=collect_dynamic_decisions, outputs=final_output)
 
 app.launch()