import gradio as gr
import os
import subprocess
import shutil
from pathlib import Path
import time

# Function to handle file upload and ingestion
def upload_and_ingest(uploaded_file):
    if uploaded_file is None:
        return "No file uploaded."
    
    try:
        # Create the pdf_docs directory if it doesn't exist
        pdf_docs_dir = "/home/tony/pdf_docs"
        os.makedirs(pdf_docs_dir, exist_ok=True)
        
        # Copy uploaded file to pdf_docs directory
        filename = os.path.basename(uploaded_file.name)
        file_path = os.path.join(pdf_docs_dir, filename)
        shutil.copy2(uploaded_file.name, file_path)
        
        # Run the ingestion script and capture output
        result = subprocess.run(
            ["python", "/home/tony/ingest.py"],
            cwd="/home/tony",
            capture_output=True, 
            text=True
        )
        
        if result.returncode == 0:
            return f"✅ File '{filename}' uploaded and ingested successfully!\n\nIngestion Output:\n{result.stdout}"
        else:
            return f"❌ Error during ingestion:\n{result.stderr}\n\nStdout:\n{result.stdout}"
            
    except Exception as e:
        return f"❌ Error: {str(e)}"

# Function to handle Google Drive folder link (placeholder for now)
def link_gdrive_folder(folder_link):
    if not folder_link or not folder_link.strip():
        return "Please provide a Google Drive folder link."
    
    # TODO: Implement Google Drive integration
    return f"🚧 Google Drive integration coming soon!\nFolder link: {folder_link}"

# Create Gradio Interface
with gr.Blocks(title="PDF Ingestion Tool", theme=gr.themes.Soft()) as demo:
    gr.Markdown("# 📚 PDF Ingestion Tool")
    gr.Markdown("Upload PDF files or link Google Drive folders to ingest into the medical knowledge base.")
    
    with gr.Tab("File Upload"):
        with gr.Row():
            file_input = gr.File(
                label="Upload PDF File",
                file_types=[".pdf"],
                type="filepath"
            )
        
        upload_btn = gr.Button("Upload & Ingest", variant="primary")
        upload_output = gr.Textbox(
            label="Ingestion Status",
            lines=10,
            max_lines=20,
            show_copy_button=True
        )
        
        upload_btn.click(
            fn=upload_and_ingest,
            inputs=[file_input],
            outputs=[upload_output],
            show_progress=True
        )
    
    with gr.Tab("Google Drive"):
        with gr.Row():
            gdrive_input = gr.Textbox(
                label="Google Drive Folder Link",
                placeholder="https://drive.google.com/drive/folders/...",
                lines=1
            )
        
        gdrive_btn = gr.Button("Link & Ingest", variant="primary")
        gdrive_output = gr.Textbox(
            label="Status",
            lines=10,
            max_lines=20,
            show_copy_button=True
        )
        
        gdrive_btn.click(
            fn=link_gdrive_folder,
            inputs=[gdrive_input],
            outputs=[gdrive_output],
            show_progress=True
        )

if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=7860)