Update app.py

app.py

@@ -11,6 +11,7 @@ from report_generator import (
 import cohere
 
 # Initialize the Cohere client using your provided API key.
+# (For production, load the API key from a secure environment variable.)
 COHERE_API_KEY = "ffwtfyfLwqSOtw65psZardfqhUxMr1h7u5vzYotI"
 co = cohere.Client(COHERE_API_KEY)
 
@@ -30,6 +31,7 @@ def cohere_parse_command(command):
             inputs=[command],
             examples=cohere_examples
         )
+        # Get the prediction for the first input.
         prediction = response.classifications[0].prediction
         return prediction
     except Exception as e:
@@ -43,19 +45,22 @@ def cohere_parse_command(command):
             return "unknown"
 
 # Global containers for pending tasks and generated files.
-task_bucket = []      # Queue for tasks.
-generated_files = []  # List of filenames generated.
+task_bucket = []      # Queue of tasks waiting to be processed.
+generated_files = []  # List of filenames for files generated by completed tasks.
 bucket_lock = threading.Lock()
 
 def process_single_task(task):
-    """Process a task with a countdown timer and update its details."""
+    """Process one task with a countdown timer and update its details."""
     print(f"[{datetime.now()}] Processing task: {task['command']}")
+    # Simulate processing delay with a countdown.
     for remaining in range(5, 0, -1):
         task["timer"] = remaining
         time.sleep(1)
+    # Use Cohere (or fallback) to classify the intent.
     intent = cohere_parse_command(task["command"])
     cmd = task["command"].lower()
 
+    # Process based on the detected intent.
     if intent == "report":
         if "csv" in cmd:
             filename = generate_csv_report()
@@ -64,6 +69,7 @@ def process_single_task(task):
             filename = generate_xlsx_report()
             task["result"] = f"XLSX report generated for '{task['command']}'"
         else:
+            # Default to PDF (with subject included)
             filename = generate_pdf_report(subject=task["command"])
             task["result"] = f"PDF report generated for '{task['command']}'"
     elif intent == "diagram":
@@ -81,9 +87,9 @@ def process_single_task(task):
     print(f"[{datetime.now()}] Task completed: {task}")
 
 def background_task_processor():
-    """Continuously checks the task bucket and processes tasks concurrently."""
+    """Continuously monitors the task bucket and spawns a new thread per task."""
     while True:
-        time.sleep(1)
+        time.sleep(1)  # Poll every second.
         task = None
         with bucket_lock:
             if task_bucket:
@@ -93,7 +99,10 @@ def background_task_processor():
             threading.Thread(target=process_single_task, args=(task,), daemon=True).start()
 
 def submit_task(command, current_tasks):
-    """Adds a new task when the user clicks Submit (command remains visible)."""
+    """
+    Called when the user clicks Submit.
+    Adds a new task (without clearing the textbox) so you can review it.
+    """
     if command.strip() == "":
         return current_tasks, generated_files, command
     new_task = {
@@ -109,19 +118,18 @@ def submit_task(command, current_tasks):
     with bucket_lock:
         task_bucket.append(new_task)
         current_tasks.append(new_task)
-    return current_tasks, generated_files, command
+    return current_tasks, generated_files, command  # Keep command in textbox.
 
 def build_tasks_html(tasks):
-    """Builds an HTML table showing task details."""
+    """Build an HTML table showing task details with checkboxes for completed tasks."""
     html = """
     <style>
-      table {width: 100%; border-collapse: collapse; margin: 20px 0; font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;}
-      th, td {border: 1px solid #555; padding: 10px; text-align: left; font-size: 14px;}
-      th {background-color: #1c2833; color: #ecf0f1;}
-      tr.Complete {background-color: #145a32; color: #ecf0f1;}
-      tr.In\\ Progress {background-color: #7d6608; color: #fdfefe;}
-      tr.Pending {background-color: #7b241c; color: #fdfefe;}
-      input[type=checkbox] {width: 18px; height: 18px; transform: scale(1.2);}
+      table {width: 100%; border-collapse: collapse; margin: 10px 0; font-family: Arial, sans-serif;}
+      th, td {border: 1px solid #ddd; padding: 8px; text-align: left; font-size: 14px;}
+      th {background-color: #f2f2f2;}
+      .Complete {background-color: #d4edda;}
+      .In\\ Progress {background-color: #fff3cd;}
+      .Pending {background-color: #f8d7da;}
     </style>
     <table>
       <tr>
@@ -141,7 +149,7 @@ def build_tasks_html(tasks):
         result = task.get("result", "")
         completed_at = task.get("completed_at", "")
         details = task.get("details", "")
-        checkbox = "<input type='checkbox' disabled " + ("checked" if status=="Complete" else "") + ">"
+        checkbox = "<input type='checkbox' disabled " + ("checked" if status == "Complete" else "") + ">"
         html += (
             f"<tr class='{status}'>"
             f"<td>{task.get('submitted_at','')}</td>"
@@ -158,80 +166,73 @@ def build_tasks_html(tasks):
     return html
 
 def refresh_ui(tasks, files):
-    """Refreshes the task list UI by rebuilding the HTML table."""
+    """Refresh the UI by rebuilding the HTML task list."""
     tasks_html = build_tasks_html(tasks)
     return tasks, files, tasks_html
 
 def main():
-    # Start background task processing.
+    # Start the background task processor.
     threading.Thread(target=background_task_processor, daemon=True).start()
     
-    with gr.Blocks(css="""
-        body {background-color: #121212; font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif; color: #fdfdfd; margin: 0; padding: 0;}
-        .gradio-container {max-width: 900px; margin: auto; padding: 20px; background-color: #1e1e1e; border-radius: 8px; box-shadow: 0px 0px 15px rgba(0,0,0,0.5);}
-        .title {text-align: center; color: #fdfdfd; margin-bottom: 20px;}
-        .section-title {color: #fdfdfd; margin-top: 30px; border-bottom: 2px solid #fdfdfd; padding-bottom: 5px;}
-    """) as demo:
-        gr.Markdown("<h1 class='title'>Advanced Multi-Task Application</h1>")
+    with gr.Blocks() as demo:
+        gr.Markdown("# Advanced Multi-Task Report & Diagram Generator")
         gr.Markdown(
-            "Enter a task command below or click one of the sample commands. For example: <i>generate a report on unemployment in the United States in 2024</i>.<br>"
-            "Tasks run concurrently and once completed, files will appear in the <b>Completed Downloads</b> section."
+            "Enter your task command below or click one of the sample commands. For example:<br>"
+            "<i>generate a report on unemployment in the United States in 2024</i><br>"
+            "The system automatically detects your intent and generates a PDF report (unless another format is specified).<br>"
+            "Multiple tasks run concurrently and you can watch their live progress below."
         )
         
         with gr.Row():
             with gr.Column(scale=8):
-                command_input = gr.Textbox(label="Task Command", placeholder="Type your command here...", lines=2)
+                command_input = gr.Textbox(label="Enter Task Command", placeholder="Type your command here...")
             with gr.Column(scale=2):
-                submit_btn = gr.Button("Submit", variant="primary")
+                submit_btn = gr.Button("Submit")
         
-        gr.Markdown("<h3 class='section-title'>Sample Commands</h3>")
+        gr.Markdown("#### Sample Commands")
         with gr.Row():
             sample1 = gr.Button("Report on US Unemployment 2024")
             sample2 = gr.Button("Diagram of Sales Data")
             sample3 = gr.Button("CSV Report of User Activity")
         
-        gr.Markdown("<h3 class='section-title'>Task List</h3>")
+        gr.Markdown("### Task List")
         tasks_html_output = gr.HTML(label="Tasks Overview")
         
-        gr.Markdown("<h3 class='section-title'>Completed Downloads</h3>")
-        file_output = gr.Files(label="Download Files", file_count="multiple")
-        
-        with gr.Row():
-            refresh_btn = gr.Button("Manual Refresh")
-            auto_refresh_toggle = gr.Checkbox(value=True, label="Auto Refresh Task List")
+        gr.Markdown("### Download Generated Files")
+        file_output = gr.Files(label="Generated Files", file_count="multiple")
         
+        # Hidden states.
         tasks_state = gr.State([])
         files_state = gr.State([])
         
+        with gr.Row():
+            refresh_btn = gr.Button("Refresh Task List")
+        
+        # When Submit is clicked, add a task.
         submit_btn.click(
             submit_task,
             inputs=[command_input, tasks_state],
             outputs=[tasks_state, files_state, command_input]
         )
+        # Sample buttons populate the command textbox.
         sample1.click(lambda: "generate a report on unemployment in the United States in 2024", None, command_input)
         sample2.click(lambda: "generate diagram of sales data", None, command_input)
         sample3.click(lambda: "generate csv report of user activity", None, command_input)
+        # Manual refresh button.
         refresh_btn.click(
             refresh_ui,
             inputs=[tasks_state, files_state],
             outputs=[tasks_state, files_state, tasks_html_output]
         )
+        # Create a hidden auto-refresh button with a fixed element ID.
         auto_refresh_btn = gr.Button("Auto Refresh", visible=False, elem_id="auto_refresh_btn")
         auto_refresh_btn.click(
             refresh_ui,
             inputs=[tasks_state, files_state],
             outputs=[tasks_state, files_state, tasks_html_output]
         )
-        gr.HTML("""
-            <script>
-              setInterval(function(){
-                if(document.getElementById('auto_refresh_toggle').checked){
-                  document.getElementById('auto_refresh_btn').click();
-                }
-              }, 5000);
-            </script>
-        """)
-        auto_refresh_toggle.elem_id = "auto_refresh_toggle"
+        # JavaScript to click the hidden auto-refresh button every 5 seconds.
+        gr.HTML("<script>setInterval(function(){document.getElementById('auto_refresh_btn').click();}, 5000);</script>")
         
     demo.launch(server_name="0.0.0.0", server_port=7860)