Update app.py

app.py

@@ -11,7 +11,6 @@ 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)
 
@@ -31,7 +30,6 @@ 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:
@@ -52,15 +50,12 @@ bucket_lock = threading.Lock()
 def process_single_task(task):
     """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.
+    # Simulate processing delay.
     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()
@@ -69,7 +64,6 @@ 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":
@@ -89,7 +83,7 @@ def process_single_task(task):
 def background_task_processor():
     """Continuously monitors the task bucket and spawns a new thread per task."""
     while True:
-        time.sleep(1)  # Poll every second.
+        time.sleep(1)
         task = None
         with bucket_lock:
             if task_bucket:
@@ -99,10 +93,7 @@ def background_task_processor():
             threading.Thread(target=process_single_task, args=(task,), daemon=True).start()
 
 def submit_task(command, current_tasks):
-    """
-    Called when the user clicks Submit.
-    Adds a new task (without clearing the textbox) so you can review it.
-    """
+    """Adds a new task when the user clicks Submit (command remains visible)."""
     if command.strip() == "":
         return current_tasks, generated_files, command
     new_task = {
@@ -118,10 +109,10 @@ 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  # Keep command in textbox.
+    return current_tasks, generated_files, command
 
 def build_tasks_html(tasks):
-    """Build an HTML table showing task details with checkboxes for completed tasks."""
+    """Builds an HTML table showing task details."""
     html = """
     <style>
       table {width: 100%; border-collapse: collapse; margin: 10px 0; font-family: Arial, sans-serif;}
@@ -166,73 +157,122 @@ def build_tasks_html(tasks):
     return html
 
 def refresh_ui(tasks, files):
-    """Refresh the UI by rebuilding the HTML task list."""
+    """Refreshes the UI by rebuilding the task list HTML."""
     tasks_html = build_tasks_html(tasks)
     return tasks, files, tasks_html
 
+def update_dropdown(files):
+    """Updates the dropdown choices for preview from the list of generated files."""
+    return files
+
+def preview_file(file_path):
+    """Returns HTML for previewing the file based on its type."""
+    if not file_path:
+        return "No file selected."
+    if file_path.endswith(".png"):
+        return f'<img src="{file_path}" style="max-width:100%;">'
+    elif file_path.endswith(".pdf"):
+        return f'<embed src="{file_path}" type="application/pdf" width="100%" height="600px">'
+    elif file_path.endswith(".csv") or file_path.endswith(".xlsx"):
+        return f'<a href="{file_path}" target="_blank">Download and view file: {file_path}</a>'
+    else:
+        return f'<a href="{file_path}" target="_blank">Download file: {file_path}</a>'
+
 def main():
-    # Start the background task processor.
     threading.Thread(target=background_task_processor, daemon=True).start()
     
-    with gr.Blocks() as demo:
-        gr.Markdown("# Advanced Multi-Task Report & Diagram Generator")
+    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>")
         gr.Markdown(
-            "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."
+            "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 complete, files will appear in the <b>Completed Downloads</b> section.<br>"
+            "You can also preview a completed file in the <b>File Preview</b> section."
         )
         
         with gr.Row():
             with gr.Column(scale=8):
-                command_input = gr.Textbox(label="Enter Task Command", placeholder="Type your command here...")
+                command_input = gr.Textbox(label="Task Command", placeholder="Type your command here...", lines=2)
             with gr.Column(scale=2):
-                submit_btn = gr.Button("Submit")
+                submit_btn = gr.Button("Submit", variant="primary")
         
-        gr.Markdown("#### Sample Commands")
+        gr.Markdown("<h3 class='section-title'>Sample Commands</h3>")
         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("### Task List")
+        gr.Markdown("<h3 class='section-title'>Task List</h3>")
         tasks_html_output = gr.HTML(label="Tasks Overview")
         
-        gr.Markdown("### Download Generated Files")
-        file_output = gr.Files(label="Generated Files", file_count="multiple")
+        gr.Markdown("<h3 class='section-title'>Completed Downloads</h3>")
+        file_output = gr.Files(label="Download Files", file_count="multiple")
         
-        # Hidden states.
-        tasks_state = gr.State([])
-        files_state = gr.State([])
+        gr.Markdown("<h3 class='section-title'>File Preview</h3>")
+        with gr.Row():
+            file_dropdown = gr.Dropdown(choices=[], label="Select File for Preview")
+            preview_btn = gr.Button("Preview File")
+        preview_output = gr.HTML(label="File Preview")
         
         with gr.Row():
-            refresh_btn = gr.Button("Refresh Task List")
+            refresh_btn = gr.Button("Manual Refresh")
+            auto_refresh_toggle = gr.Checkbox(value=True, label="Auto Refresh Task List")
+        
+        tasks_state = gr.State([])
+        files_state = gr.State([])
         
-        # 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.
+        # Hidden auto-refresh button.
         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]
         )
-        # JavaScript to click the hidden auto-refresh button every 5 seconds.
-        gr.HTML("<script>setInterval(function(){document.getElementById('auto_refresh_btn').click();}, 5000);</script>")
+        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"
+        
+        # Update the file preview dropdown when refreshing tasks.
+        refresh_btn.click(
+            update_dropdown,
+            inputs=[files_state],
+            outputs=[file_dropdown]
+        )
+        auto_refresh_btn.click(
+            update_dropdown,
+            inputs=[files_state],
+            outputs=[file_dropdown]
+        )
+        # Preview button: shows the preview for the selected file.
+        preview_btn.click(
+            preview_file,
+            inputs=[file_dropdown],
+            outputs=[preview_output]
+        )
         
     demo.launch(server_name="0.0.0.0", server_port=7860)