Create app.py

app.py

@@ -0,0 +1,115 @@
+import os
+import firebase_admin
+from firebase_admin import credentials, db
+from transformers import pipeline
+import gradio as gr
+
+# ----------------------------------
+# Initialize Firebase
+# ----------------------------------
+# Replace with your Firebase credentials
+firebase_cred_path = "path/to/your-serviceAccountKey.json"  # Upload to your Hugging Face space
+firebase_db_url = "https://your-database-name.firebaseio.com/"  # Replace with your Realtime Database URL
+
+cred = credentials.Certificate(firebase_cred_path)
+firebase_admin.initialize_app(cred, {"databaseURL": firebase_db_url})
+
+# ----------------------------------
+# Initialize Hugging Face RAG Model
+# ----------------------------------
+rag_model = pipeline("question-answering", model="facebook/rag-token-base")
+
+
+# ----------------------------------
+# Functions for Firebase Operations
+# ----------------------------------
+def fetch_tasks():
+    """Fetch tasks from Firebase Realtime Database."""
+    ref = db.reference("projects")
+    data = ref.get()
+    if not data:
+        return "No tasks found!"
+    output = ""
+    for project_id, project_data in data.items():
+        output += f"Project: {project_id}\n"
+        for task_id, task_data in project_data.get("tasks", {}).items():
+            output += f"- Task: {task_data['title']} (Status: {task_data['status']})\n"
+    return output
+
+
+def add_task(project_id, task_name, description, assignee, deadline):
+    """Add a new task to Firebase."""
+    ref = db.reference(f"projects/{project_id}/tasks")
+    new_task_ref = ref.push()  # Create a new task
+    new_task_ref.set(
+        {
+            "title": task_name,
+            "description": description,
+            "status": "in-progress",
+            "assignee": assignee,
+            "deadline": deadline,
+        }
+    )
+    return f"Task '{task_name}' added to project '{project_id}' successfully!"
+
+
+def query_rag(question):
+    """Query RAG model with project data as context."""
+    ref = db.reference("projects")
+    data = ref.get()
+    if not data:
+        return "No data available for RAG."
+
+    # Prepare context from Firebase data
+    context = ""
+    for project_id, project_data in data.items():
+        for task_id, task_data in project_data.get("tasks", {}).items():
+            context += f"{task_data['title']}: {task_data['status']} (Assigned to: {task_data['assignee']}).\n"
+
+    # Query RAG model
+    response = rag_model(question=question, context=context)
+    return response["answer"]
+
+
+# ----------------------------------
+# Gradio Dashboard
+# ----------------------------------
+def dashboard():
+    """Gradio interface for Project Management Dashboard."""
+    with gr.Blocks() as app:
+        gr.Markdown("## Project Management Dashboard with Live Updates")
+        
+        # Add Task Section
+        with gr.Tab("Add Task"):
+            project_id = gr.Textbox(label="Project ID")
+            task_name = gr.Textbox(label="Task Name")
+            description = gr.Textbox(label="Description")
+            assignee = gr.Textbox(label="Assignee")
+            deadline = gr.Textbox(label="Deadline (YYYY-MM-DD)")
+            add_task_button = gr.Button("Add Task")
+            add_task_output = gr.Textbox(label="Output", interactive=False)
+            add_task_button.click(
+                add_task,
+                [project_id, task_name, description, assignee, deadline],
+                add_task_output,
+            )
+
+        # Fetch Tasks Section
+        with gr.Tab("View Tasks"):
+            fetch_button = gr.Button("Fetch Tasks")
+            tasks_output = gr.Textbox(label="Tasks", interactive=False)
+            fetch_button.click(fetch_tasks, [], tasks_output)
+
+        # Query RAG Section
+        with gr.Tab("Ask Questions"):
+            question = gr.Textbox(label="Ask a question about your projects")
+            query_button = gr.Button("Get Answer")
+            rag_output = gr.Textbox(label="Answer", interactive=False)
+            query_button.click(query_rag, [question], rag_output)
+
+    app.launch()
+
+
+# Launch the app
+if __name__ == "__main__":
+    dashboard()