Update app.py

app.py

@@ -1,58 +1,74 @@
+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
 
-# Initialize Firebase
-cred = credentials.Certificate("firebase-key.json")
-firebase_admin.initialize_app(cred)
-db = firestore.client()
+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):
-    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()
-        # Add to Firebase
-        db.collection("tasks").add({
-            "task": task,
-            "created": current_time,
-            "status": "pending"
-        })
-        response = f"✅ Task added: {task}\nCreated at: {current_time}"
-    
-    elif message == "/list":
-        # Get from Firebase
-        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
+   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])
+   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])
 
-demo.launch()
+if __name__ == "__main__":
+   demo.launch()