"""app.py — Gradio + Gemini + apify-client. Version 8.0.0 | 4 April 2026. ZERO for/while/if."""
import os
import glob
import gradio as gr
from langchain_google_genai import ChatGoogleGenerativeAI
from langgraph.prebuilt import create_react_agent
from langgraph.checkpoint.memory import MemorySaver
from agent import SYSTEM_PROMPT, get_local_tools

print(">>> app.py: imports complete")

llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash", temperature=0)
tools = get_local_tools()
agent = create_react_agent(model=llm, tools=tools, prompt=SYSTEM_PROMPT, checkpointer=MemorySaver())
print(f">>> app.py: agent ready ({len(tools)} tools)")

_msg_count = 0

HEADER_HTML = """
<style>
@import url('https://fonts.googleapis.com/css2?family=DM+Sans:wght@400;500;600;700&display=swap');
.gradio-container {max-width: 960px !important; margin: 0 auto !important; font-family: 'DM Sans', sans-serif !important;}
footer {display: none !important;}
</style>
<div style="background: linear-gradient(135deg, #1a1a2e 0%, #16213e 50%, #0f3460 100%); border-radius: 12px; padding: 16px 24px; margin-bottom: 12px; color: white;">
    <h2 style="margin: 0 0 4px 0; font-size: 1.3em; font-weight: 700; color: #e0e0ff;">🤖 Agentic AI — Smart Web Scraping Agent</h2>
    <p style="margin: 0; font-size: 0.8em; color: #94a3b8; line-height: 1.4;">Powered by <b>Gemini 2.5 Flash</b> + <b>22,000+ Apify Actors</b> + <b>apify-client SDK</b></p>
    <div style="margin-top: 6px;">
        <span style="display:inline-block;background:rgba(255,255,255,0.1);border-radius:20px;padding:2px 10px;font-size:0.72em;margin:2px 3px 2px 0;color:#c4b5fd;">TripAdvisor</span>
        <span style="display:inline-block;background:rgba(255,255,255,0.1);border-radius:20px;padding:2px 10px;font-size:0.72em;margin:2px 3px 2px 0;color:#c4b5fd;">Instagram</span>
        <span style="display:inline-block;background:rgba(255,255,255,0.1);border-radius:20px;padding:2px 10px;font-size:0.72em;margin:2px 3px 2px 0;color:#c4b5fd;">Google Maps</span>
        <span style="display:inline-block;background:rgba(255,255,255,0.1);border-radius:20px;padding:2px 10px;font-size:0.72em;margin:2px 3px 2px 0;color:#c4b5fd;">Amazon</span>
        <span style="display:inline-block;background:rgba(255,255,255,0.1);border-radius:20px;padding:2px 10px;font-size:0.72em;margin:2px 3px 2px 0;color:#c4b5fd;">Any Website</span>
        <span style="display:inline-block;background:rgba(255,255,255,0.1);border-radius:20px;padding:2px 10px;font-size:0.72em;margin:2px 3px 2px 0;color:#c4b5fd;">22K+ more</span>
    </div>
</div>
"""


def _latest_file():
    """Find most recent CSV in /tmp."""
    files = sorted(glob.glob("/tmp/*.csv"), key=os.path.getmtime)
    return (files and files[-1]) or None


def respond(message, chat_history):
    """Fresh thread per message. CSV auto-downloads from last scrape."""
    global _msg_count
    _msg_count += 1
    text = (message or "").strip() or "Search Apify for popular web scrapers"
    print(f"\n{'='*60}\n>>> MESSAGE #{_msg_count}: '{text}'\n{'='*60}")

    chat_history = chat_history + [
        {"role": "user", "content": text},
        {"role": "assistant", "content": "🔍 **Searching & scraping...**\n\nFinding actor → reading schema → running scraper → analyzing.\n\n_Please wait 30-90 seconds._"},
    ]
    yield chat_history, "", _latest_file()

    result = agent.invoke(
        {"messages": [("human", text)]},
        config={"configurable": {"thread_id": f"t{_msg_count}"}},
    )
    response = result["messages"][-1].content
    print(f">>> Response ({len(response)} chars)")

    chat_history[-1] = {"role": "assistant", "content": response}
    yield chat_history, "", _latest_file()


print(">>> Building UI...")
with gr.Blocks(title="Agentic AI - Web Scraping Agent") as demo:
    gr.HTML(HEADER_HTML)
    chatbot = gr.Chatbot(height=340, show_label=False)
    with gr.Row():
        msg = gr.Textbox(placeholder="Scrape TripAdvisor reviews for Taj Dubai · Get 10 Instagram posts for Nike · Amazon reviews for LG TV",
            show_label=False, scale=9, lines=1, max_lines=2, container=False)
        send = gr.Button("⏎", variant="primary", scale=1, min_width=50)
    download = gr.File(label="📥 Download CSV", visible=True)
    msg.submit(respond, [msg, chatbot], [chatbot, msg, download])
    send.click(respond, [msg, chatbot], [chatbot, msg, download])

print(">>> Launching...")
demo.launch(server_name="0.0.0.0", server_port=7860)