import gradio as gr
from utils.wikipedia_fetcher import fetch_wikipedia_content
from ai.debate_generator import generate_debate, load_model
from utils.translator import translate_text
from ai.debate_generator import model, tokenizer

# Use a small model for Spaces
import ai.debate_generator
ai.debate_generator.MODEL_NAME = "distilgpt2"
load_model()

def debate_interface(topic, lang):
    wiki_text = fetch_wikipedia_content(topic)
    debate = generate_debate(wiki_text, model, tokenizer)
    if lang != "en":
        debate = translate_text(debate, lang)
    return debate

iface = gr.Interface(
    fn=debate_interface,
    inputs=[
        gr.Textbox(label="Debate Topic", placeholder="Enter a topic..."),
        gr.Dropdown(["en", "hi", "fr"], value="en", label="Language")
    ],
    outputs=gr.Textbox(label="Debate Output"),
    title="WikiDebate",
    description="Debate anything, powered by facts — not opinions. Generates AI-powered pro vs con debates on controversial or curious topics using data from Wikipedia, in any language."
)

if __name__ == "__main__":
    iface.launch()