Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 from deep_translator import GoogleTranslator
 import torch
 
-# انتخاب مدل پایدار و سازگار
+# مدل پیشنهادی پایدار
 model_id = "google/gemma-7b-it"
 
 # بارگذاری توکنایزر و مدل
@@ -17,7 +17,6 @@ model.eval()
 
 # تابع تولید موضوع پایان‌نامه
 def generate_topics(field, major, keywords, audience, level):
-    # ساخت پرامپت برای مدل
     prompt = f"""<bos>[INST]Suggest 3 academic thesis topics based on the following:
 Field: {field}
 Specialization: {major}
@@ -25,26 +24,27 @@ Keywords: {keywords}
 Target Audience: {audience}
 Level: {level}[/INST]"""
 
-    # توکن‌سازی و ارسال به مدل
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
     with torch.no_grad():
         outputs = model.generate(**inputs, max_new_tokens=256)
-    english_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    english_output = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
 
-    # ترجمه خروجی به فارسی
-    translated_output = GoogleTranslator(source='en', target='fa').translate(english_output)
-    translated_output_html = translated_output.strip().replace("\n", "<br>")
+    if not english_output:
+        return "<div dir='rtl' style='color: red;'>❌ مشکلی در تولید موضوعات رخ داد. لطفاً دوباره تلاش کنید.</div>"
+
+    translated_output = GoogleTranslator(source='en', target='fa').translate(english_output).strip()
+    translated_output_html = translated_output.replace("\n", "<br>")
 
-    # ساخت خروجی HTML راست‌چین برای Gradio
     html_output = (
-        "<div dir='rtl' style='text-align: right; font-family: Tahoma, sans-serif; font-size: 16px; line-height: 1.8;'>"
+        "<div dir='rtl' style='text-align: right; font-family: Tahoma, sans-serif; "
+        "font-size: 16px; line-height: 1.8; min-height: 250px;'>"
         f"{translated_output_html}"
         "<br><br>📢 برای مشاوره و راهنمایی تخصصی با گروه مشاوره کاسپین تماس بگیرید:<br>"
         "<strong>021-88252497</strong></div>"
     )
     return html_output
 
-# رابط کاربری با Gradio
+# رابط Gradio
 iface = gr.Interface(
     fn=generate_topics,
     inputs=[
@@ -55,7 +55,8 @@ iface = gr.Interface(
         gr.Dropdown(choices=["کارشناسی ارشد", "دکتری"], label="مقطع")
     ],
     outputs=gr.HTML(label="موضوعات پیشنهادی"),
-    title="🎓 پیشنهادگر موضوع پایان‌نامه کاسپین"
+    title="🎓 پیشنهادگر موضوع پایان‌نامه کاسپین",
+    theme="default"
 )
 
 iface.launch()