import gradio as gr
from openai import OpenAI
import os

# قراءة مفتاح API من المتغيرات السرية
client = OpenAI(
    base_url="https://openrouter.ai/api/v1",
    api_key=os.environ.get("OPENROUTER_API_KEY")
)

available_models = [
    "openrouter/owl-alpha",
    "meta-llama/llama-3.3-70b-instruct:free",
    "minimax/minimax-m2.5:free",
    "openai/gpt-oss-120b:free",
    "nvidia/nemotron-3-super-120b-a12b:free",
    "qwen/qwen3-next-80b-a3b-instruct:free",
    "liquid/lfm-2.5-1.2b-thinking:free",
    "arcee-ai/trinity-large-thinking:free",
    "openrouter/free",
]

system_prompt = """
أنت مساعد ذكي لحملة "التوقف عن جوجل".
مهمتك: تقديم بدائل حرة لخدمات جوجل، والإجابة على أي سؤال متعلق بالخصوصية الرقمية.
تتحدث كل اللغات، خاصة العربية واللهجة المصرية العامية، وتجاوب بطريقة ودودة ومباشرة.
أنشأك المبرمج المصري "عبده" قائد الحملة.
"""

def chat_with_abdo(message, history, model_choice):
    messages = [{"role": "system", "content": system_prompt}]
    for user_msg, bot_msg in history:
        messages.append({"role": "user", "content": user_msg})
        messages.append({"role": "assistant", "content": bot_msg})
    messages.append({"role": "user", "content": message})

    response = client.chat.completions.create(
        model=model_choice,
        messages=messages,
        temperature=0.7,
        max_tokens=500
    )
    return response.choices[0].message.content

iface = gr.ChatInterface(
    fn=chat_with_abdo,
    additional_inputs=[
        gr.Dropdown(choices=available_models, value=available_models[0], label="اختر الموديل")
    ],
    title="عبده AI - مساعد حملة التوقف عن جوجل",
    description="أهلاً بيك! إسأل عن أي بديل لخدمات جوجل. أفهم كل اللغات واللهجة المصرية 🇪🇬"
)

iface.launch(server_name="0.0.0.0", server_port=7860, share=True)