from workflow.agents_workflow import build_agents_workflow
from workflow.graph_state import GraphState

# Run locally without gradio

app = build_agents_workflow(llm_model_name="gpt-4o-mini")

def run_user_query(ticker):
    QUERY = f"Produce investor-ready insights for {ticker}."
    init_state: GraphState = {
        "ticker": ticker,
        "query": QUERY,
        "news_summary": None,
        "earnings_summary": None,
        "market_summary": None,
        "completed": [],
        "final_recommendation": None,
    }
    final_state = app.invoke(init_state)

    return final_state

state = run_user_query("AAPL")

print("\n" + "=" * 80)
print(f"### NEWS SUMMARY\n{state['news_summary']}\n")
print(f"### EARNINGS SUMMARY\n{state['earnings_summary']}\n")
print(f"### MARKET SUMMARY\n{state['market_summary']}\n")
print(f"### FINAL RECOMMENDATION\n{state['final_recommendation']}\n")