fix empty correction

README.md

@@ -10,9 +10,13 @@ pinned: true
 license: mit 
 ---
 
-# ✅ Write4All
+# ✅Write4All
 
-In a world where diversity and inclusivity are essential, crafting content that resonates with everyone can be a challenge. With ✅ Write4All I hope making the world more inclusive, one word and image at a time.
+In a world where diversity and inclusivity are essential, crafting content that resonates with everyone can be a challenge. With ✅Write4All I hope making the world more inclusive, one word and image at a time.
+
+See it in action in:
+- HuggingFace Spaces: [✅Write4All](https://huggingface.co/spaces/Jaume/Write4All).
+- Examples notebook: [✅Write4All Examples](./notebooks/examples.ipynb)
 
 ## Requirements
 

review.py

@@ -46,13 +46,15 @@ gemini_1_0_vision = genai.GenerativeModel(
 
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
+
 @cache
 def get_file(relative_path: str) -> str:
     current_path = os.path.dirname(os.path.abspath(__file__))
     full_path = os.path.join(current_path, relative_path)
     with open(full_path) as f:
         return f.read()
-    
+
+
 def fix_json(json_str: str) -> str:
     template = get_file("templates/prompt_json_fix.txt")
     prompt = template.format(json=json_str)
@@ -90,9 +92,14 @@ def apply_review(text: str, review: list[dict]) -> str:
             start = starts[0]
             end = start + len(entity["term"])
             output += text[last_end:start]
-            output += get_file("templates/correction.html").format(
-                term=text[start:end], fix=entity["fix"], kind=entity["type"]
-            )
+            if len(entity["fix"]) > 0:
+                output += get_file("templates/correction.html").format(
+                    term=text[start:end], fix=entity["fix"], kind=entity["type"]
+                )
+            else:
+                output += get_file("templates/deletion.html").format(
+                    term=text[start:end], kind=entity["type"]
+                )
             last_end = end
     output += text[last_end:]
     return f"<pre style='white-space: pre-wrap;'>{output}</pre>"
@@ -105,6 +112,7 @@ def review_table_summary(review: list[dict]) -> str:
     table += "</table>"
     return table
 
+
 def review_text(text: str, text_model: genai.GenerativeModel) -> list[dict]:
     template = get_file("templates/prompt_v1.txt")
     try:
@@ -116,6 +124,7 @@ def review_text(text: str, text_model: genai.GenerativeModel) -> list[dict]:
         )
     return get_json_content(response)
 
+
 def process_text(model: str, text: str) -> str:
     text_model = gemini_1_0 if model == "Gemini 1.0 Pro" else gemini_1_5
     review = review_text(text, text_model)
@@ -142,4 +151,4 @@ def review_image(image, vision_model: genai.GenerativeModel) -> list[dict]:
 
 def process_image(model: str, image):
     vision_model = gemini_1_0_vision if model == "Gemini 1.0 Pro Vision" else gemini_1_5
-    return review_image(image, vision_model)
+    return review_image(image, vision_model)

templates/deletion.html

@@ -0,0 +1 @@
+<span title="{kind}"><span style='background-color:#FF5555; color:black; text-decoration:line-through; text-decoration-thickness: 2px; min-height: 22px; display: inline-block; border-radius: 5px 5px 5px 5px; padding-left: 5px; padding-right: 5px;'>{term}</span></span>