import os
import gradio as gr
from transformers import pipeline
from argostranslate import package, translate

# تولید مدل انگلیسی برای موضوع پایان‌نامه
generator = pipeline("text-generation", model="tiiuae/falcon-rw-1b")

# گردان ترجمه انگلیسی به فارسی (argos-translate)
installed_languages = translate.get_installed_languages()
en_lang = list(filter(lambda x: x.code == "en", installed_languages))[0]
fa_lang = list(filter(lambda x: x.code == "fa", installed_languages))[0]
translator = en_lang.get_translation(fa_lang)

def generate_topics(field, major, keywords, audience, level):
    prompt = f"""
Suggest 3 academic thesis topics based on the following information:
Field: {field}
Specialization: {major}
Keywords: {keywords}
Target audience: {audience}
Level: {level}
"""

    # generate English topics
    english_output = generator(prompt, max_new_tokens=256)[0]['generated_text']

    # translate to Persian
    translated_output = translator.translate(english_output)

    # add final message
    final_output = translated_output.strip() + "\n\n📢 برای مشاوره و راهنمایی تخصصی با گروه مشاوره کاسپین تماس بگیرید:\n02188252497"

    return final_output

iface = gr.Interface(
    fn=generate_topics,
    inputs=[
        gr.Textbox(label="رشته"),
        gr.Textbox(label="گرایش"),
        gr.Textbox(label="کلیدواژه‌ها"),
        gr.Textbox(label="جامعه هدف"),
        gr.Dropdown(choices=["کارشناسی ارشد", "دکتری"], label="مقطع")
    ],
    outputs="text",
    title="🎓 پیشنهادگر موضوع پایان‌نامه کاسپین"
)

iface.launch()