fix bug of showing forensic

src/application/formatting_fact_checker.py

@@ -5,6 +5,7 @@ from src.application.formatting import (
     color_text,
     format_entity_count,
 )
+from src.application.image.helper import encode_image
 from src.application.image.image import ImageDetector
 from src.application.text.entity import apply_highlight
 from src.application.text.helper import (
@@ -238,15 +239,22 @@ def format_image_fact_checker_row(image: ImageDetector):
         return ""
 
     if image.referent_url is not None or image.referent_url != "":
-        source_image = f"""<img src="{image.referent_url}" width="100" height="150">"""  # noqa: E501
+        if "http" in image.input:
+            input_image = (
+                f"""<a href="{image.input}">{image.input}</a>"""  # noqa: E501
+            )
+        else:
+            base64_image = encode_image(image.input)
+            input_image = f"""<img src="data:image/jpeg;base64,{base64_image}" width="100" height="150">"""  # noqa: E501
         source_image_url = f"""<a href="{image.referent_url}">{image.referent_url}</a>"""  # noqa: E501
+        source_image = f"""<img src="{image.referent_url}" width="100" height="150">"""  # noqa: E501
     else:
         source_image = "Image not found"
         source_image_url = ""
 
     return f"""
 <tr>
-    <td>input image</td>
+    <td>{input_image}</td>
     <td>{source_image}</td>
     <td>{image.prediction_label}<br>({image.prediction_score:.2f}%)</td>
     <td style="{WORD_BREAK}";>{source_image_url}</td>

src/application/formatting_governor.py

@@ -5,6 +5,7 @@ from src.application.formatting import (
     color_text,
     format_entity_count,
 )
+from src.application.image.helper import encode_image
 from src.application.image.image import ImageDetector
 from src.application.text.entity import apply_highlight
 from src.application.text.helper import (
@@ -155,20 +156,27 @@ def format_text_governor_row(text):
 """
 
 
-def format_image_governor_row(image):
+def format_image_governor_row(image: ImageDetector):
     if image.input is None:
         return ""
 
     if image.referent_url is not None or image.referent_url != "":
-        source_image = f"""<img src="{image.referent_url}" width="100" height="150">"""  # noqa: E501
+        if "http" in image.input:
+            input_image = (
+                f"""<a href="{image.input}">{image.input}</a>"""  # noqa: E501
+            )
+        else:
+            base64_image = encode_image(image.input)
+            input_image = f"""<img src="data:image/jpeg;base64,{base64_image}" width="100" height="150">"""  # noqa: E501
         source_image_url = f"""<a href="{image.referent_url}">{image.referent_url}</a>"""  # noqa: E501
+        source_image = f"""<img src="{image.referent_url}" width="100" height="150">"""  # noqa: E501
     else:
         source_image = "Image not found"
         source_image_url = ""
 
     return f"""
 <tr>
-    <td>input image</td>
+    <td>{input_image}</td>
     <td>{source_image}</td>
     <td>{image.prediction_label}<br>({image.prediction_score:.2f}%)</td>
     <td style="{WORD_BREAK}";>{source_image_url}</td>

src/application/formatting_ordinary_user.py

@@ -1,6 +1,7 @@
 from pandas import DataFrame
 
 from src.application.config import WORD_BREAK
+from src.application.image.helper import encode_image
 from src.application.image.image import ImageDetector
 from src.application.text.helper import replace_leading_spaces
 from src.application.text.text import TextDetector
@@ -105,13 +106,20 @@ def format_image_ordinary_user_row(image: ImageDetector) -> str:
 
     # Put image, label, and score into html tag
     if image.referent_url is not None or image.referent_url != "":
+        if "http" in image.input:
+            input_image = (
+                f"""<a href="{image.input}">{image.input}</a>"""  # noqa: E501
+            )
+        else:
+            base64_image = encode_image(image.input)
+            input_image = f"""<img src="data:image/jpeg;base64,{base64_image}" width="100" height="150">"""  # noqa: E501
         source_image_html = f"""<a href="{image.referent_url}">{image.referent_url}</a>"""  # noqa: E501
     else:
         source_image_html = ""
 
     return f"""
 <tr>
-    <td>input image</td>
+    <td>{input_image}</td>
     <td>{image.prediction_label}<br>({image.prediction_score:.2f}%)</td>
     <td style="{WORD_BREAK}";>{source_image_html}</td>
 </tr>

src/application/image/helper.py

@@ -1,12 +1,24 @@
 import base64
 
 
-def encode_image(image_path):
-    with open(image_path, "rb") as img_file:
-        return base64.b64encode(img_file.read()).decode()
+def encode_image(image_path: str) -> str:
+    """
+    Encodes an image file into a Base64 string.
 
+    Args:
+        image_path (str): The path to the image file.
 
-image_base64 = encode_image(
-    "/content/ai-generated-picture-of-a-tiger-walking-in-the-forest-photo.jpg",
-)
-html_code = f'<img src="data:image/jpeg;base64,{image_base64}" width="300">'
+    Returns:
+        str: The Base64 encoded string representation of the image.
+
+    Raises:
+        FileNotFoundError: If the specified image file does not exist.
+        IOError: If there is an error reading the image file.
+    """
+    try:
+        with open(image_path, "rb") as img_file:
+            return base64.b64encode(img_file.read()).decode()
+    except FileNotFoundError:
+        raise FileNotFoundError(f"Image file not found at: {image_path}")
+    except OSError as e:
+        raise OSError(f"Error reading image file: {e}")