import gradio as gr
from dotenv import load_dotenv
from deep_research_agents.research_manager import ResearchManager
import sendgrid
from sendgrid.helpers.mail import Email, Mail, Content, To
import os
import ssl
ssl._create_default_https_context = ssl._create_unverified_context

load_dotenv(override=True)

# Custom email sender to use the user's email
async def send_email_to_user(subject: str, html_body: str, user_email: str):
    sg = sendgrid.SendGridAPIClient(api_key=os.environ.get('SENDGRID_API_KEY'))
    from_email = Email("liorsolomon.solid@gmail.com")  # put your verified sender here
    to_email = To(user_email)
    content = Content("text/html", html_body)
    mail = Mail(from_email, to_email, subject, content).get()
    response = sg.client.mail.send.post(request_body=mail)
    print("Email response", response.status_code)
    return {"status": "success"}

# Patch the ResearchManager to allow dynamic recipient email
def make_research_manager_with_email(user_email):
    class PatchedResearchManager(ResearchManager):
        async def send_email(self, report):
            subject = "Your Deep Research Report"
            html_body = report.markdown_report.replace('\n', '<br>')
            await send_email_to_user(subject, html_body, user_email)
            return report
    return PatchedResearchManager()

async def run_research(query: str, user_email: str):
    manager = make_research_manager_with_email(user_email)
    yield "<b>🔍 Research started! Please wait while we process your request...</b>"
    async for chunk in manager.run(query):
        # Show step-by-step progress to the user
        if chunk.startswith("View trace: "):
            yield f"<i>Trace available: <a href='{chunk[11:]}' target='_blank'>{chunk[11:]}</a></i>"
        elif chunk.lower().startswith("searches planned"):
            yield "<b>📝 Searches planned. Starting to search the web...</b>"
        elif chunk.lower().startswith("searches complete"):
            yield "<b>✅ Searches complete. Writing the report...</b>"
        elif chunk.lower().startswith("report written"):
            yield "<b>🖊️ Report written. Sending the report to your email...</b>"
        elif chunk.lower().startswith("email sent"):
            yield "<b>📧 Email sent! Research complete. See your inbox for the full report.</b>"
        elif chunk.strip().startswith('#'):
            yield chunk
        else:
            yield f"<i>{chunk}</i>"

with gr.Blocks(theme=gr.themes.Default(primary_hue="sky")) as ui:
    gr.Markdown("# Deep Research Chatbot\nEnter your email and research topic. You'll receive a detailed report by email!")
    with gr.Row():
        email_box = gr.Textbox(label="Your Email", type="text", placeholder="your@email.com")
        query_box = gr.Textbox(label="Research Query", lines=3, placeholder="What would you like to research?")
    run_btn = gr.Button("Run Deep Research", variant="primary")
    report_md = gr.Markdown(label="Report Preview (sent to your email)")

    run_btn.click(run_research, inputs=[query_box, email_box], outputs=report_md)
    query_box.submit(run_research, inputs=[query_box, email_box], outputs=report_md)
    email_box.submit(run_research, inputs=[query_box, email_box], outputs=report_md)

if __name__ == "__main__":
    ui.launch(inbrowser=True)