import gradio as gr
from inference import get_evo_response, get_gpt_response
from logger import log_feedback

def advisor_interface(query, context, feedback_choice):
    evo_output = get_evo_response(query, context)
    gpt_output = get_gpt_response(query, context)

    if feedback_choice != "No feedback":
        log_feedback(query, context, evo_output, feedback_choice)

    return evo_output, gpt_output

with gr.Blocks() as demo:
    gr.Markdown("## 🧠 EvoAdvisor – Your Adaptive AI Copilot for Finance")

    with gr.Row():
        query = gr.Textbox(label="📝 Ask a financial question", placeholder="e.g. Should we exit this position?")
        context = gr.Textbox(label="📂 Related memo or policy context", placeholder="Paste internal strategy or notes here...")

    with gr.Row():
        feedback = gr.Radio(["👍 Helpful", "👎 Not Helpful", "No feedback"], label="Was Evo’s answer useful?", value="No feedback")

    with gr.Row():
        evo_out = gr.Textbox(label="🔬 EvoAdvisor's Suggestion")
        gpt_out = gr.Textbox(label="🤖 GPT-3.5's Suggestion")

    submit_btn = gr.Button("Run Advisors")
    submit_btn.click(fn=advisor_interface, inputs=[query, context, feedback], outputs=[evo_out, gpt_out])

demo.launch()