from smolagents import CodeAgent, HfApiModel
import gradio as gr
import os

api_key = os.getenv("hf_key")


agent = CodeAgent(
    tools=[],
    model=HfApiModel(token=api_key),
    add_base_tools=True,
    verbosity_level=2,
)

def eval1():
    return agent.run(" باللغة العربية ضع خطة لتطوير وزارة التعليم السعودي")

def eval2():
    return agent.run(" باللغة العربية ضع خطة لتطوير وزارة الاعلام السعودي")

def eval3():
    return agent.run(" باللغة العربية ضع خطة لتطوير وزارة الاتصالات السعودي")

def eval4():
    return agent.run(" باللغة العربية ضع خطة لتطوير وزارة الزراعة السعودي")

with gr.Blocks() as demo:
    with gr.Row(equal_height=True):
        greet_btn1 = gr.Button("ضع خطة لتطوير وزارة التعليم السعودي", scale=1, variant='stop')
        greet_btn2 = gr.Button("ضع خطة لتطوير وزارة الاعلام السعودي", scale=1, variant='huggingface')
        greet_btn3 = gr.Button("ضع خطة لتطوير وزارة الاتصالات السعودي", scale=1, variant='secondary')
        greet_btn4 = gr.Button("ضع خطة لتطوير وزارة الزراعة السعودي", scale=1, variant='primary')

    output = gr.Textbox(label="النتائج")

    greet_btn1.click(fn=eval1, outputs=output, api_name="ضع خطة لتطوير وزارة التعليم السعودي")
    greet_btn2.click(fn=eval2, outputs=output, api_name="ضع خطة لتطوير وزارة الاعلام السعودي")
    greet_btn3.click(fn=eval3, outputs=output, api_name="ضع خطة لتطوير وزارة الاتصالات السعودي")
    greet_btn4.click(fn=eval4, outputs=output, api_name="ضع خطة لتطوير وزارة الزراعة السعودي")


if __name__ == "__main__":
    demo.launch(share=True)