import gradio as gr
from model_wrapper import CodeDebuggerWrapper

# instantiate once (will download model)
debugger = CodeDebuggerWrapper()

def debug_code(code: str):
    if not code or not code.strip():
        return "❌ Please paste some code to debug."
    try:
        return debugger.debug(code)
    except Exception as e:
        # friendly error message
        return f"Error during model inference:\n{e}"

with gr.Blocks() as demo:
    gr.Markdown("## 🐞 AI Code Debugger (Fine-tuned)")
    gr.Markdown("Paste Python code below and click **Debug Code**. Uses your fine-tuned HF model.")
    code_input = gr.Textbox(lines=14, placeholder="Paste Python code here...", label="Input Code")
    output = gr.Textbox(lines=14, label="Debugged Code / Output")
    btn = gr.Button("Debug Code")
    btn.click(fn=debug_code, inputs=code_input, outputs=output)

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