Update app.py

app.py

@@ -5,17 +5,19 @@ import io
 from PIL import Image
 import numpy as np
 from zhipuai import ZhipuAI
+import os
 
-# Thay API Key của bạn tại đây
-API_KEY = "d659608f7d5d42b1821a9303fc50b618.NesCuIytwpxjxpBJ"
-
-# Khởi tạo ZhipuAI client
+# Lấy API Key từ biến môi trường
+API_KEY = os.getenv("ZHIPUAI_API_KEY", "d659608f7d5d42b1821a9303fc50b618.NesCuIytwpxjxpBJ")
 client = ZhipuAI(api_key=API_KEY)
 
 def extract_text(image):
     if isinstance(image, np.ndarray):  # Chuyển NumPy array thành ảnh
         image = Image.fromarray(image)
 
+    # Giới hạn kích thước ảnh
+    image = image.resize((min(image.width, 1024), min(image.height, 1024)), Image.Resampling.LANCZOS)
+
     # Chuyển đổi ảnh sang Base64
     buffered = io.BytesIO()
     image.save(buffered, format="PNG")
@@ -47,7 +49,8 @@ def extract_text(image):
 
         return text, output_text_file, output_image_file
     else:
-        return "Lỗi: Không nhận được kết quả từ API.", None, None
+        error_msg = f"Lỗi: {str(response)}" if response else "Lỗi: Không kết nối được API."
+        return error_msg, None, None
 
 demo = gr.Interface(
     fn=extract_text,
@@ -62,4 +65,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()