Update app.py

app.py

@@ -1,53 +1,29 @@
-from transformers import pipeline
-import gradio as gr
-
-# Tạo pipeline với mô hình GPT-2 tiếng Việt từ NlpHUST
-text_generator = pipeline(
-    "text-generation",
-    model="NlpHUST/gpt2-vietnamese",
-    tokenizer="NlpHUST/gpt2-vietnamese"
-)
-
-# Hàm xử lý sinh bio
 def generate_bio(prompt, max_length, temperature):
-    # Làm sạch prompt và chuẩn hóa định dạng
     prompt = prompt.strip()
     if not prompt:
         return "❗️Vui lòng nhập thông tin mô tả bản thân để tạo bio."
-    
-    full_prompt = f"Viết một đoạn giới thiệu bản thân hấp dẫn cho mạng xã hội:\n{prompt}\n\nBio:"
-    
-    # Sinh văn bản
+
+    full_prompt = f"Viết một đoạn bio ngắn, giới thiệu bản thân với nội dung:\n{prompt}\n\nBio:"
+
     try:
         outputs = text_generator(
             full_prompt,
-            max_length=max_length,
+            max_length=min(max_length, 80),  # Giới hạn cứng tránh vượt quá
             temperature=temperature,
             do_sample=True,
             top_k=50,
             top_p=0.95,
-            num_return_sequences=1
+            num_return_sequences=1,
+            eos_token_id=text_generator.tokenizer.eos_token_id  # Dừng đúng chỗ
         )
+        # Cắt kết quả đúng phần sinh ra
         result = outputs[0]["generated_text"].replace(full_prompt, "").strip()
+        # Giới hạn độ dài đầu ra
+        if len(result) > max_length:
+            result = result[:max_length].rsplit('.', 1)[0] + '.'
         return result if result else "⚠️ Không tạo được bio phù hợp. Hãy thử lại với mô tả khác."
     except Exception as e:
         return f"🚫 Lỗi xảy ra khi sinh văn bản: {str(e)}"
 
-# Giao diện Gradio
-gr.Interface(
-    fn=generate_bio,
-    inputs=[
-        gr.Textbox(label="🧠 Mô tả bản thân / điểm mạnh", placeholder="VD: Tôi là một chuyên gia marketing đam mê giáo dục...", lines=4),
-        gr.Slider(50, 300, value=150, step=10, label="📝 Độ dài tối đa"),
-        gr.Slider(0.1, 1.0, value=0.8, step=0.05, label="🎨 Mức sáng tạo (temperature)")
-    ],
-    outputs=gr.Textbox(label="✅ Đoạn bio gợi ý", lines=5),
-    title="✨ AI Viết Bio Thương Hiệu Cá Nhân (Tiếng Việt)",
-    description="Nhập mô tả bản thân, AI sẽ gợi ý đoạn bio phù hợp cho mạng xã hội, trang cá nhân hoặc phần giới thiệu nghề nghiệp.",
-    theme="default"
-).launch()
-
-
-