import os
import gradio as gr
import firebase_admin
from firebase_admin import credentials, firestore
from datetime import datetime
from dotenv import load_dotenv
from pathlib import Path

load_dotenv()

BASE_DIR = Path(__file__).resolve().parent
FIREBASE_KEY = BASE_DIR / 'firebase-key.json'

try:
   cred = credentials.Certificate(str(FIREBASE_KEY))
   firebase_admin.initialize_app(cred)
   db = firestore.client()
except Exception as e:
   print(f"Firebase initialization error: {e}")

def add_task(message, history):
   try:
       if not message:
           return "", history
       
       history = history or []
       current_time = datetime.now().strftime("%Y-%m-%d %H:%M")
       
       if message.startswith("/task"):
           task = message[6:].strip()
           if not task:
               return "", history + [("", "Task description required")]
           
           db.collection("tasks").add({
               "task": task,
               "created": current_time,
               "status": "pending"
           })
           response = f"✅ Task added: {task}\nCreated at: {current_time}"
           
       elif message == "/list":
           tasks_ref = db.collection("tasks").stream()
           tasks = [task.to_dict() for task in tasks_ref]
           
           if not tasks:
               response = "No tasks found."
           else:
               response = "📋 Tasks:\n" + "\n".join([
                   f"{i+1}. {task['task']} ({task['status']}) - {task['created']}"
                   for i, task in enumerate(tasks)
               ])
       else:
           response = "Commands:\n/task [description] - Add new task\n/list - View all tasks"
       
       history.append((message, response))
       return "", history
       
   except Exception as e:
       return "", history + [("", f"Error: {str(e)}")]

with gr.Blocks() as demo:
   gr.Markdown("# 📝 TaskMate")
   gr.Markdown("### Task Management Made Simple")
   
   chatbot = gr.Chatbot(height=400)
   msg = gr.Textbox(
       placeholder="Type /task [description] to add a task, or /list to view tasks",
       label="Input"
   )
   
   msg.submit(add_task, [msg, chatbot], [msg, chatbot])

if __name__ == "__main__":
   demo.launch()