Update app.py

app.py

@@ -5,6 +5,7 @@ import re
 import gradio as gr
 import google.generativeai as genai
 import random
+from PIL import Image # Thêm dòng này
 
 print("Google Generative AI SDK version:", genai.__version__)
 
@@ -70,15 +71,34 @@ def process_image(image, prompt):
         #     print("(Info) RAG store exists:", e)
 
         # Create temp file
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp:
+        COMPRESSION_QUALITY = 80
+
+        # === KHỐI CODE THAY THẾ ===
+        # Tạo tệp tạm thời với suffix là .jpeg
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".jpeg") as tmp:
+            
+            # 1. Mở hình ảnh bằng Pillow (xử lý các loại input khác nhau)
             if hasattr(image, "save"):
-                image.save(tmp.name)
-            elif isinstance(image, str):
-                with open(image, "rb") as f:
-                    tmp.write(f.read())
+                # Trường hợp input đã là đối tượng Pillow Image
+                img = image
+            elif isinstance(image, str) and os.path.exists(image):
+                # Trường hợp input là đường dẫn tệp
+                img = Image.open(image)
             else:
+                # Trường hợp input không được hỗ trợ
                 raise ValueError(f"Unsupported image type: {type(image)}")
+            
+            # 2. Giảm Chất Lượng (quality=80) và Lưu tệp vào tệp tạm thời
+            # Lưu dưới định dạng JPEG để tối ưu hóa việc nén
+            img.save(
+                tmp.name, 
+                format='JPEG', 
+                optimize=True, 
+                quality=COMPRESSION_QUALITY
+            )
+            
             temp_file = tmp.name
+        # ===========================
         print("Temp file:", temp_file)
 
         # Upload file