import gradio as gr
import requests
import os

# Load environment variables from Hugging Face Secrets
MATH_ENGINE_URL = "https://abhlash-mathengine.hf.space/calculate"

# Gradio Interface
history = []
current_input = ""

def gradio_calculate(a, b, operation):
    url = MATH_ENGINE_URL
    payload = {"a": a, "b": b, "operation": operation}
    headers = {"Content-Type": "application/json"}
    response = requests.post(url, json=payload, headers=headers)
    result = response.json().get("result", "Error: " + response.json().get("detail", "Unknown error"))
    history.append(f"{a} {operation} {b} = {result}")
    return "\n".join(history)

def update_input(digit):
    global current_input
    current_input += str(digit)
    return current_input

def clear_input():
    global current_input
    current_input = ""
    return current_input

def backspace():
    global current_input
    current_input = current_input[:-1]
    return current_input

def test_fastapi(a, b, operation):
    url = MATH_ENGINE_URL
    payload = {"a": a, "b": b, "operation": operation}
    headers = {"Content-Type": "application/json"}
    response = requests.post(url, json=payload, headers=headers)
    return response.json()

def create_interface():
    with gr.Blocks() as demo:
        a = gr.Number(label="A", interactive=True)
        b = gr.Number(label="B", interactive=True)
        result = gr.Textbox(label="Result", lines=10, interactive=False)
        current = gr.Textbox(label="Current Input", interactive=False)

        with gr.Row():
            for i in range(10):
                gr.Button(str(i)).click(lambda x=i: update_input(x), inputs=[], outputs=current)
            gr.Button("Back").click(backspace, inputs=[], outputs=current)
            gr.Button("Clear").click(clear_input, inputs=[], outputs=current)

        with gr.Row():
            gr.Button("Set A").click(lambda x: float(current_input), inputs=[], outputs=a)
            gr.Button("Set B").click(lambda x: float(current_input), inputs=[], outputs=b)

        with gr.Row():
            gr.Button("Add").click(lambda a, b: gradio_calculate(a, b, "add"), inputs=[a, b], outputs=result)
            gr.Button("Subtract").click(lambda a, b: gradio_calculate(a, b, "subtract"), inputs=[a, b], outputs=result)
            gr.Button("Multiply").click(lambda a, b: gradio_calculate(a, b, "multiply"), inputs=[a, b], outputs=result)
            gr.Button("Divide").click(lambda a, b: gradio_calculate(a, b, "divide"), inputs=[a, b], outputs=result)
            gr.Button("Power").click(lambda a, b: gradio_calculate(a, b, "power"), inputs=[a, b], outputs=result)
            gr.Button("Modulus").click(lambda a, b: gradio_calculate(a, b, "modulus"), inputs=[a, b], outputs=result)
            gr.Button("Square Root").click(lambda a: gradio_calculate(a, None, "sqrt"), inputs=[a], outputs=result)
            gr.Button("Log").click(lambda a: gradio_calculate(a, None, "log"), inputs=[a], outputs=result)
            gr.Button("Log10").click(lambda a: gradio_calculate(a, None, "log10"), inputs=[a], outputs=result)
            gr.Button("Sin").click(lambda a: gradio_calculate(a, None, "sin"), inputs=[a], outputs=result)
            gr.Button("Cos").click(lambda a: gradio_calculate(a, None, "cos"), inputs=[a], outputs=result)
            gr.Button("Tan").click(lambda a: gradio_calculate(a, None, "tan"), inputs=[a], outputs=result)
            gr.Button("Asin").click(lambda a: gradio_calculate(a, None, "asin"), inputs=[a], outputs=result)
            gr.Button("Acos").click(lambda a: gradio_calculate(a, None, "acos"), inputs=[a], outputs=result)
            gr.Button("Atan").click(lambda a: gradio_calculate(a, None, "atan"), inputs=[a], outputs=result)
            gr.Button("Factorial").click(lambda a: gradio_calculate(a, None, "factorial"), inputs=[a], outputs=result)

    return demo

def run_gradio():
    demo = create_interface()
    demo.launch()

if __name__ == "__main__":
    run_gradio()