Create app.py

app.py

@@ -0,0 +1,135 @@
+import gradio as gr
+import re
+import textwrap
+from io import BytesIO
+
+# Các thư viện cho việc tạo ảnh từ văn bản
+from PIL import Image, ImageDraw, ImageFont
+import matplotlib.pyplot as plt
+
+
+def render_plain_text_image(text, font_size, width, height, bg_color, text_color):
+    """
+    Render văn bản thông thường (không chứa LaTeX) thành ảnh bằng Pillow.
+    Nếu chiều cao (height) do người dùng nhập vào không đủ để chứa text, 
+    chương trình sẽ tự động điều chỉnh lại (với căn giữa theo chiều dọc).
+    """
+    # Thử load font TrueType; nếu không có, dùng font mặc định
+    try:
+        font = ImageFont.truetype("arial.ttf", font_size)
+    except Exception:
+        font = ImageFont.load_default()
+
+    # Tạo ảnh tạm để đo kích thước của text
+    temp_img = Image.new("RGB", (width, 100), color=bg_color)
+    draw = ImageDraw.Draw(temp_img)
+    # Ước tính chiều rộng của một ký tự (giả sử "A")
+    char_width, _ = draw.textsize("A", font=font)
+    # Số ký tự tối đa trên 1 dòng ước tính từ chiều rộng ảnh
+    max_chars = max(1, width // char_width)
+    # Tự động ngắt dòng theo số ký tự
+    lines = textwrap.wrap(text, width=max_chars)
+    
+    # Tính tổng chiều cao cần dùng cho các dòng text
+    line_heights = []
+    for line in lines:
+        _, lh = draw.textsize(line, font=font)
+        line_heights.append(lh)
+    total_text_height = sum(line_heights)
+    
+    # Nếu chiều cao được truyền vào không đủ, ta dùng chiều cao tối thiểu
+    if height is None or height < total_text_height + 20:
+        height = total_text_height + 20
+
+    # Tạo ảnh nền với kích thước (width x height)
+    img = Image.new("RGB", (width, height), color=bg_color)
+    draw = ImageDraw.Draw(img)
+    # Căn giữa theo chiều dọc: tính lề trên
+    y_text = (height - total_text_height) // 2
+    for line in lines:
+        line_width, line_height = draw.textsize(line, font=font)
+        # Căn giữa theo chiều ngang
+        x_text = (width - line_width) // 2
+        draw.text((x_text, y_text), line, font=font, fill=text_color)
+        y_text += line_height
+    return img
+
+
+def render_math_image(text, font_size, width, height, bg_color, text_color):
+    """
+    Render văn bản chứa LaTeX (có dấu $ … $) thành ảnh bằng matplotlib.
+    Chúng ta dùng matplotlib để hiển thị công thức (mathtext) theo kiểu toán học.
+    """
+    dpi = 100
+    # Chuyển kích thước từ pixel sang inch (1 inch = dpi pixel)
+    fig_width = width / dpi
+    fig_height = height / dpi
+    fig = plt.figure(figsize=(fig_width, fig_height), dpi=dpi)
+    ax = plt.axes([0, 0, 1, 1])
+    ax.axis('off')
+    fig.patch.set_facecolor(bg_color)
+    # Dùng hàm text() với thuộc tính wrap=True và căn giữa
+    ax.text(0.5, 0.5, text, color=text_color, fontsize=font_size,
+            ha='center', va='center', wrap=True)
+    buf = BytesIO()
+    # Lưu figure vào bộ nhớ với một chút padding
+    plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0.1)
+    plt.close(fig)
+    buf.seek(0)
+    from PIL import Image
+    img = Image.open(buf)
+    return img
+
+
+def text_to_image(file, text, font_size, width, height, bg_color, text_color):
+    """
+    Hàm chính: nếu người dùng upload file thì đọc text từ file; nếu không thì dùng nội dung
+    nhập trực tiếp. Nếu văn bản có chứa định dạng LaTeX (dấu $ … $) thì dùng hàm render_math_image,
+    ngược lại dùng render_plain_text_image.
+    """
+    # Nếu file được upload, ưu tiên sử dụng nội dung file
+    if file is not None:
+        try:
+            # Trong Gradio, file đầu vào có thể là dict chứa key "name"
+            if isinstance(file, dict):
+                file_path = file["name"]
+            else:
+                file_path = file
+            with open(file_path, 'r', encoding='utf-8') as f:
+                text = f.read()
+        except Exception as e:
+            return f"Error reading file: {e}"
+    if not text:
+        return "Chưa có nội dung text được nhập!"
+
+    # Nếu phát hiện chuỗi LaTeX (có dấu $ … $) thì dùng mode math
+    if re.search(r'\$(.*?)\$', text):
+        return render_math_image(text, font_size, width, height, bg_color, text_color)
+    else:
+        return render_plain_text_image(text, font_size, width, height, bg_color, text_color)
+
+
+# Xây dựng giao diện Gradio
+iface = gr.Interface(
+    fn=text_to_image,
+    inputs=[
+        gr.File(label="Upload file chứa text (tùy chọn)"),
+        gr.Textbox(label="Nhập text", lines=10, placeholder="Nhập text của bạn tại đây..."),
+        gr.Slider(10, 100, step=1, value=20, label="Font Size"),
+        gr.Slider(200, 2000, step=10, value=800, label="Chiều rộng ảnh (pixel)"),
+        gr.Slider(100, 2000, step=10, value=600, label="Chiều cao ảnh (pixel)"),
+        gr.ColorPicker(value="#ffffff", label="Màu nền"),
+        gr.ColorPicker(value="#000000", label="Màu chữ")
+    ],
+    outputs=gr.Image(type="pil"),
+    title="Text-to-Image Converter",
+    description=(
+        "Upload một file chứa text hoặc nhập trực tiếp text. "
+        "Nếu text có chứa biểu thức LaTeX (ví dụ: $\\int_a^b f(x)dx$) "
+        "thì nó sẽ được render theo kiểu toán (math). "
+        "Các tuỳ chọn bên dưới cho phép bạn điều chỉnh đầu ra."
+    )
+)
+
+if __name__ == "__main__":
+    iface.launch()