Create ui/manual_dashboard.py

ui/manual_dashboard.py

@@ -0,0 +1,363 @@
+"""
+Manual Dashboard UI - Visual tool interface
+"""
+
+import gradio as gr
+import asyncio
+from pathlib import Path
+from typing import List
+import json
+
+
+def create_manual_dashboard(agent):
+    """Create manual dashboard interface"""
+    
+    # State for uploaded files
+    uploaded_files_state = gr.State([])
+    
+    with gr.Row():
+        # Left sidebar - File upload and management
+        with gr.Column(scale=1):
+            gr.Markdown("### 📁 File Manager")
+            
+            file_upload = gr.File(
+                label="Upload Files",
+                file_count="multiple",
+                file_types=[".pdf", ".png", ".jpg", ".jpeg", ".docx", ".txt", ".csv"]
+            )
+            
+            files_list = gr.Textbox(
+                label="Uploaded Files",
+                placeholder="No files uploaded yet",
+                interactive=False,
+                lines=5
+            )
+            
+            gr.Markdown("---")
+            gr.Markdown("### 🔧 Quick Actions")
+            
+            with gr.Group():
+                ocr_btn = gr.Button("🔍 Extract Text (OCR)", variant="secondary", size="sm")
+                summarize_btn = gr.Button("📝 Summarize Document", variant="secondary", size="sm")
+                metadata_btn = gr.Button("📊 Extract Metadata", variant="secondary", size="sm")
+                calendar_btn = gr.Button("📅 Create Calendar Event", variant="secondary", size="sm")
+                email_btn = gr.Button("✉️ Draft Email", variant="secondary", size="sm")
+                organize_btn = gr.Button("🗂️ Organize Files", variant="secondary", size="sm")
+        
+        # Main content area
+        with gr.Column(scale=2):
+            gr.Markdown("### 🎯 Tool Operations")
+            
+            # Tabbed interface for different tools
+            with gr.Tabs():
+                # OCR Tool
+                with gr.Tab("🔍 OCR"):
+                    with gr.Row():
+                        ocr_file_select = gr.Dropdown(
+                            label="Select File",
+                            choices=[],
+                            interactive=True
+                        )
+                        ocr_language = gr.Dropdown(
+                            label="Language",
+                            choices=["en", "es", "fr", "de", "zh"],
+                            value="en"
+                        )
+                    
+                    ocr_execute_btn = gr.Button("Extract Text", variant="primary")
+                    
+                    ocr_output = gr.Textbox(
+                        label="Extracted Text",
+                        lines=10,
+                        placeholder="Text will appear here..."
+                    )
+                    
+                    ocr_confidence = gr.Textbox(label="Confidence Score")
+                
+                # PDF Summarization Tool
+                with gr.Tab("📝 Summarize"):
+                    pdf_file_select = gr.Dropdown(
+                        label="Select PDF",
+                        choices=[],
+                        interactive=True
+                    )
+                    
+                    summary_length = gr.Slider(
+                        label="Summary Length (words)",
+                        minimum=100,
+                        maximum=1000,
+                        value=300,
+                        step=50
+                    )
+                    
+                    summarize_execute_btn = gr.Button("Generate Summary", variant="primary")
+                    
+                    summary_output = gr.Textbox(
+                        label="Summary",
+                        lines=10,
+                        placeholder="Summary will appear here..."
+                    )
+                    
+                    pdf_metadata = gr.JSON(label="Document Metadata")
+                
+                # Calendar Tool
+                with gr.Tab("📅 Calendar"):
+                    with gr.Row():
+                        event_title = gr.Textbox(label="Event Title", placeholder="Meeting with team")
+                        event_location = gr.Textbox(label="Location (optional)", placeholder="Conference Room A")
+                    
+                    with gr.Row():
+                        event_start = gr.Textbox(
+                            label="Start Date/Time",
+                            placeholder="2024-12-01 10:00 or Dec 1, 2024 10:00 AM"
+                        )
+                        event_end = gr.Textbox(
+                            label="End Date/Time",
+                            placeholder="2024-12-01 11:00"
+                        )
+                    
+                    event_description = gr.Textbox(
+                        label="Description",
+                        lines=3,
+                        placeholder="Event details..."
+                    )
+                    
+                    calendar_execute_btn = gr.Button("Create Event", variant="primary")
+                    
+                    calendar_output = gr.File(label="Download ICS File")
+                    calendar_status = gr.Textbox(label="Status")
+                
+                # Email Drafting Tool
+                with gr.Tab("✉️ Email"):
+                    with gr.Row():
+                        email_recipient = gr.Textbox(label="Recipient", placeholder="john@example.com")
+                        email_subject = gr.Textbox(label="Subject", placeholder="Regarding...")
+                    
+                    email_context = gr.Textbox(
+                        label="Context/Purpose",
+                        lines=4,
+                        placeholder="What should this email be about?"
+                    )
+                    
+                    email_tone = gr.Radio(
+                        label="Tone",
+                        choices=["professional", "friendly", "formal", "casual"],
+                        value="professional"
+                    )
+                    
+                    email_execute_btn = gr.Button("Draft Email", variant="primary")
+                    
+                    email_output = gr.Textbox(
+                        label="Draft Email",
+                        lines=10,
+                        placeholder="Email draft will appear here..."
+                    )
+                    
+                    email_download = gr.File(label="Download Email Files")
+                
+                # Form Filler Tool
+                with gr.Tab("📋 Form Filler"):
+                    form_template = gr.File(label="Upload Form Template (.docx or .xlsx)")
+                    
+                    form_data = gr.Textbox(
+                        label="Form Data (JSON)",
+                        lines=8,
+                        placeholder='{\n  "name": "John Doe",\n  "email": "john@example.com",\n  "date": "2024-12-01"\n}',
+                        value='{}'
+                    )
+                    
+                    form_execute_btn = gr.Button("Fill Form", variant="primary")
+                    
+                    form_output = gr.File(label="Download Filled Form")
+                    form_status = gr.Textbox(label="Status")
+                
+                # File Organizer Tool
+                with gr.Tab("🗂️ Organize"):
+                    organize_strategy = gr.Radio(
+                        label="Organization Strategy",
+                        choices=["by_type", "by_date", "by_size"],
+                        value="by_type"
+                    )
+                    
+                    organize_execute_btn = gr.Button("Organize Files", variant="primary")
+                    
+                    organize_output = gr.JSON(label="Organization Results")
+    
+    # Event handlers
+    async def handle_file_upload(files):
+        if not files:
+            return "No files uploaded", [], []
+        
+        file_list = []
+        file_info = []
+        
+        for file in files:
+            # Copy to uploads directory
+            from utils.file_utils import copy_file, get_file_info
+            
+            dest_path = f"data/uploads/{Path(file.name).name}"
+            copy_file(file.name, dest_path)
+            
+            info = get_file_info(dest_path)
+            file_list.append(dest_path)
+            file_info.append(f"✓ {info['name']} ({info['size_mb']} MB)")
+            
+            # Add to RAG
+            await agent.process_files_to_rag([{'path': dest_path, 'name': info['name']}])
+        
+        files_text = "\n".join(file_info)
+        return files_text, file_list, file_list
+    
+    async def handle_ocr(file_path, language):
+        if not file_path:
+            return "Please select a file", ""
+        
+        result = await agent.manual_tool_call(
+            'ocr_extract_text',
+            {'file_path': file_path, 'language': language}
+        )
+        
+        if result['success']:
+            data = result['result']
+            return data.get('text', ''), f"Confidence: {data.get('confidence', 0):.2%}"
+        else:
+            return f"Error: {result['error']}", ""
+    
+    async def handle_summarize(file_path, max_length):
+        if not file_path:
+            return "Please select a file", {}
+        
+        result = await agent.manual_tool_call(
+            'pdf_summarize',
+            {'file_path': file_path, 'max_length': max_length}
+        )
+        
+        if result['success']:
+            data = result['result']
+            return data.get('summary', ''), data.get('metadata', {})
+        else:
+            return f"Error: {result['error']}", {}
+    
+    async def handle_calendar(title, start, end, description, location):
+        result = await agent.manual_tool_call(
+            'calendar_create_event',
+            {
+                'title': title,
+                'start': start,
+                'end': end,
+                'description': description,
+                'location': location
+            }
+        )
+        
+        if result['success']:
+            data = result['result']
+            return data.get('output_path'), f"✓ Event created: {title}"
+        else:
+            return None, f"✗ Error: {result['error']}"
+    
+    async def handle_email(recipient, subject, context, tone):
+        result = await agent.manual_tool_call(
+            'email_draft',
+            {
+                'recipient': recipient,
+                'subject': subject,
+                'context': context,
+                'tone': tone
+            }
+        )
+        
+        if result['success']:
+            data = result['result']
+            return data.get('email_body', ''), data.get('output_path_html')
+        else:
+            return f"Error: {result['error']}", None
+    
+    async def handle_form_fill(template_file, data_json):
+        if not template_file:
+            return None, "Please upload a form template"
+        
+        try:
+            data = json.loads(data_json)
+        except:
+            return None, "Invalid JSON data"
+        
+        result = await agent.manual_tool_call(
+            'form_fill',
+            {
+                'template_path': template_file.name,
+                'data': data
+            }
+        )
+        
+        if result['success']:
+            res_data = result['result']
+            return res_data.get('output_path'), f"✓ Form filled: {len(res_data.get('fields_filled', []))} fields"
+        else:
+            return None, f"✗ Error: {result['error']}"
+    
+    async def handle_organize(file_list, strategy):
+        if not file_list:
+            return {"error": "No files to organize"}
+        
+        # Use upload directory
+        result = await agent.manual_tool_call(
+            'file_organize',
+            {
+                'directory': 'data/uploads',
+                'strategy': strategy
+            }
+        )
+        
+        if result['success']:
+            return result['result']
+        else:
+            return {"error": result['error']}
+    
+    # Wire up events
+    file_upload.change(
+        fn=handle_file_upload,
+        inputs=[file_upload],
+        outputs=[files_list, uploaded_files_state, ocr_file_select, pdf_file_select]
+    )
+    
+    ocr_execute_btn.click(
+        fn=handle_ocr,
+        inputs=[ocr_file_select, ocr_language],
+        outputs=[ocr_output, ocr_confidence]
+    )
+    
+    ocr_btn.click(
+        fn=lambda: gr.Tabs.update(selected="🔍 OCR"),
+        outputs=[]
+    )
+    
+    summarize_execute_btn.click(
+        fn=handle_summarize,
+        inputs=[pdf_file_select, summary_length],
+        outputs=[summary_output, pdf_metadata]
+    )
+    
+    calendar_execute_btn.click(
+        fn=handle_calendar,
+        inputs=[event_title, event_start, event_end, event_description, event_location],
+        outputs=[calendar_output, calendar_status]
+    )
+    
+    email_execute_btn.click(
+        fn=handle_email,
+        inputs=[email_recipient, email_subject, email_context, email_tone],
+        outputs=[email_output, email_download]
+    )
+    
+    form_execute_btn.click(
+        fn=handle_form_fill,
+        inputs=[form_template, form_data],
+        outputs=[form_output, form_status]
+    )
+    
+    organize_execute_btn.click(
+        fn=handle_organize,
+        inputs=[uploaded_files_state, organize_strategy],
+        outputs=[organize_output]
+    )