Create app.py

app.py

@@ -0,0 +1,300 @@
+import gradio as gr
+import os
+import time
+import json
+from datetime import datetime
+import random
+import string
+
+def main():
+    with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange")) as demo:
+        # Header
+        gr.Markdown("# AI Synthesis Manager")
+        
+        # ID-based process handling
+        with gr.Row():
+            with gr.Column(scale=2):
+                # New synthesis section
+                gr.Markdown("## Create New Synthesis")
+                query_input = gr.Textbox(
+                    label="Enter Query:",
+                    placeholder="What topic would you like to generate a synthesis for?",
+                    lines=2
+                )
+                
+                with gr.Row():
+                    create_button = gr.Button("Generate New Synthesis ID", variant="primary")
+                    synthesis_id = gr.Textbox(label="Synthesis ID", placeholder="ID will appear here...")
+            
+            with gr.Column(scale=2):
+                # Continue existing synthesis section
+                gr.Markdown("## Continue Existing Synthesis")
+                existing_id_input = gr.Textbox(
+                    label="Enter Existing Synthesis ID:",
+                    placeholder="Enter ID to continue from a previous session..."
+                )
+                
+                with gr.Row():
+                    load_button = gr.Button("Load Synthesis State")
+        
+        # Synthesis status display
+        with gr.Row(visible=False) as status_panel:
+            gr.Markdown("## Current Synthesis Status")
+            
+            with gr.Column():
+                status_query = gr.Textbox(label="Query")
+                status_id = gr.Textbox(label="Synthesis ID")
+                
+                with gr.Row():
+                    status_created = gr.Textbox(label="Created", scale=1)
+                    status_updated = gr.Textbox(label="Last Updated", scale=1)
+                
+                with gr.Row():
+                    status_progress = gr.Slider(
+                        minimum=0, maximum=8, step=1, value=3,
+                        label="Progress (Steps Completed)"
+                    )
+                    status_cost = gr.Number(label="Cost So Far ($)", value=0.00)
+        
+        # Step Status Cards
+        steps_container = gr.Row(visible=False)
+        with steps_container:
+            gr.Markdown("## Pipeline Steps")
+            
+            with gr.Column():
+                # Step 1: Preoutline
+                with gr.Group():
+                    gr.Markdown("### 1. Preoutline")
+                    step1_status = gr.Markdown("✅ **Completed**")
+                    with gr.Row():
+                        step1_view = gr.Button("View Results", size="sm")
+                        step1_download = gr.Button("Download", size="sm")
+                
+                # Step 2: Themes Generator
+                with gr.Group():
+                    gr.Markdown("### 2. Themes Generator")
+                    step2_status = gr.Markdown("✅ **Completed**")
+                    with gr.Row():
+                        step2_view = gr.Button("View Results", size="sm")
+                        step2_download = gr.Button("Download", size="sm")
+                
+                # Step 3: Outline
+                with gr.Group():
+                    gr.Markdown("### 3. Outline")
+                    step3_status = gr.Markdown("❌ **Failed** (Connection timeout)")
+                    with gr.Row():
+                        step3_retry = gr.Button("Retry Step", size="sm", variant="primary")
+                        step3_view = gr.Button("View Partial Results", size="sm")
+                        step3_download = gr.Button("Download Partial", size="sm")
+                
+                # Step 4: Research
+                with gr.Group():
+                    gr.Markdown("### 4. Research")
+                    step4_status = gr.Markdown("⏳ **Waiting** (Depends on Outline)")
+                    with gr.Row():
+                        step4_force = gr.Button("Force Start", size="sm")
+            
+            with gr.Column():
+                # Step 5: Thinking Ladder
+                with gr.Group():
+                    gr.Markdown("### 5. Thinking Ladder")
+                    step5_status = gr.Markdown("⏳ **Waiting** (Depends on Research)")
+                    with gr.Row():
+                        step5_force = gr.Button("Force Start", size="sm")
+                
+                # Step 6: Enhanced Outline
+                with gr.Group():
+                    gr.Markdown("### 6. Enhanced Outline")
+                    step6_status = gr.Markdown("⏳ **Waiting** (Depends on Thinking Ladder)")
+                    with gr.Row():
+                        step6_force = gr.Button("Force Start", size="sm")
+                
+                # Step 7: Synthesis
+                with gr.Group():
+                    gr.Markdown("### 7. Synthesis")
+                    step7_status = gr.Markdown("⏳ **Waiting** (Depends on Enhanced Outline)")
+                    with gr.Row():
+                        step7_force = gr.Button("Force Start", size="sm")
+                
+                # Step 8: Accountability
+                with gr.Group():
+                    gr.Markdown("### 8. Accountability")
+                    step8_status = gr.Markdown("⏳ **Waiting** (Depends on Synthesis)")
+                    with gr.Row():
+                        step8_force = gr.Button("Force Start", size="sm")
+        
+        # Result Preview
+        with gr.Row(visible=False) as result_panel:
+            gr.Markdown("## Results Preview")
+            # Removed 'interactive=False' as it's not a valid parameter for gr.JSON
+            result_json = gr.JSON(label="Synthesis Results")
+            
+            with gr.Row():
+                refresh_results = gr.Button("Refresh Results")
+                download_final = gr.Button("Download Final Results")
+        
+        # Log panel
+        with gr.Row(visible=False) as log_panel:
+            gr.Markdown("## Process Log")
+            log_output = gr.Textbox(
+                label="", 
+                lines=8,
+                max_lines=15,
+                placeholder="Process logs will appear here..."
+            )
+        
+        # Mock function for generating a new synthesis ID
+        def generate_new_id(query):
+            if not query.strip():
+                return "Please enter a query first", gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), ""
+            
+            # Generate a random ID (in real implementation, this would be more robust)
+            synthesis_id = ''.join(random.choices(string.ascii_uppercase + string.digits, k=8))
+            
+            # Update UI
+            timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            return (
+                synthesis_id,
+                gr.update(visible=True),
+                gr.update(visible=True), 
+                gr.update(visible=True),
+                f"[{timestamp}] Created new synthesis with ID: {synthesis_id}\n[{timestamp}] Starting preoutline generation for query: {query}...\n",
+                query,
+                synthesis_id,
+                datetime.now().strftime("%Y-%m-%d %H:%M"),
+                datetime.now().strftime("%Y-%m-%d %H:%M"),
+                0.00
+            )
+        
+        # Mock function for loading existing synthesis
+        def load_existing(existing_id):
+            if not existing_id.strip():
+                return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), "Please enter a valid synthesis ID"
+            
+            # In a real implementation, this would fetch data from AWS based on the ID
+            # For demonstration, we'll use mock data
+            
+            # Update UI components
+            timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            
+            return (
+                gr.update(visible=True),
+                gr.update(visible=True),
+                gr.update(visible=True),
+                f"[{timestamp}] Loaded synthesis with ID: {existing_id}\n[{timestamp}] Last step completed: Themes Generator\n[{timestamp}] Next step: Outline (previously failed)\n",
+                "Sustainability in urban planning",
+                existing_id,
+                "2025-04-25 14:30",
+                "2025-04-25 14:45",
+                2,
+                0.85
+            )
+        
+        # Connect the handlers
+        create_button.click(
+            fn=generate_new_id,
+            inputs=[query_input],
+            outputs=[
+                synthesis_id, 
+                status_panel, 
+                steps_container, 
+                log_panel, 
+                log_output,
+                status_query,
+                status_id,
+                status_created,
+                status_updated,
+                status_cost
+            ]
+        )
+        
+        load_button.click(
+            fn=load_existing,
+            inputs=[existing_id_input],
+            outputs=[
+                status_panel, 
+                steps_container, 
+                log_panel, 
+                log_output,
+                status_query,
+                status_id,
+                status_created,
+                status_updated,
+                status_progress,
+                status_cost
+            ]
+        )
+        
+        # Mock function for step retry
+        def retry_step(id):
+            timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            return (
+                "🟡 **In Progress**...",
+                f"[{timestamp}] Retrying Outline step for synthesis ID: {id}\n[{timestamp}] Connecting to AWS...\n"
+            )
+        
+        # Connect retry button
+        step3_retry.click(
+            fn=retry_step,
+            inputs=[status_id],
+            outputs=[step3_status, log_output]
+        )
+        
+        # Mock function for force start
+        def force_start(step_name, id):
+            timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            return (
+                "🟡 **Force Started**...",
+                f"[{timestamp}] Force starting {step_name} step for synthesis ID: {id}\n[{timestamp}] Warning: This step depends on previous incomplete steps\n"
+            )
+        
+        # Connect force start buttons
+        step4_force.click(
+            fn=lambda id: force_start("Research", id),
+            inputs=[status_id],
+            outputs=[step4_status, log_output]
+        )
+        
+        step5_force.click(
+            fn=lambda id: force_start("Thinking Ladder", id),
+            inputs=[status_id],
+            outputs=[step5_status, log_output]
+        )
+        
+        # Mock view results for steps
+        def view_step_results(step_name, id):
+            mock_results = {
+                "Preoutline": {
+                    "main_points": ["Sustainable urban development", "Green infrastructure", "Smart city technologies"],
+                    "key_concepts": ["Energy efficiency", "Waste management", "Public transportation"],
+                    "potential_structure": "Introduction → Current challenges → Sustainable solutions → Case studies → Conclusion"
+                },
+                "Themes": {
+                    "themes": ["Environmental Impact", "Economic Considerations", "Social Aspects", "Technological Integration"],
+                    "query": "Sustainability in urban planning",
+                    "focus_areas": ["Renewable energy in cities", "Waste reduction strategies", "Community involvement"]
+                }
+            }
+            
+            if step_name in mock_results:
+                return gr.update(visible=True, value=mock_results[step_name])
+            return gr.update(visible=True, value={"message": f"No results available for {step_name} step"})
+        
+        # Connect view buttons
+        step1_view.click(
+            fn=lambda: view_step_results("Preoutline", ""),
+            inputs=[],
+            outputs=[result_panel]
+        )
+        
+        step2_view.click(
+            fn=lambda: view_step_results("Themes", ""),
+            inputs=[],
+            outputs=[result_panel]
+        )
+
+    return demo
+
+if __name__ == "__main__":
+    demo = main()
+    demo.launch(debug=True)