Add app demo of BLUX-cA

app.py

@@ -20,7 +20,7 @@ Run:
 """
 
 from __future__ import annotations
-import json, re, time, uuid, logging
+import json, re, time, uuid, logging, tempfile, shutil
 from dataclasses import dataclass, asdict
 from pathlib import Path
 from typing import Dict, List, Tuple, Optional, Any
@@ -272,13 +272,31 @@ class SessionManager:
     def __init__(self):
         self.session_id = str(uuid.uuid4())
         self.start_time = time.time()
+        self.temp_dir = Path(tempfile.mkdtemp(prefix="blux_ca_"))
+        logger.info(f"Session temp directory: {self.temp_dir}")
     
     def get_session_info(self) -> Dict[str, Any]:
         return {
             "session_id": self.session_id,
             "start_time": self.start_time,
-            "duration": time.time() - self.start_time
+            "duration": round(time.time() - self.start_time, 2),
+            "temp_dir": str(self.temp_dir)
         }
+    
+    def create_temp_audit_file(self) -> Path:
+        """Create a temporary audit file that Gradio can safely serve."""
+        audit_file = self.temp_dir / f"blux_ca_audit_{self.session_id}.jsonl"
+        audit_file.touch()
+        return audit_file
+    
+    def cleanup(self):
+        """Clean up temporary files."""
+        try:
+            if self.temp_dir.exists():
+                shutil.rmtree(self.temp_dir)
+                logger.info(f"Cleaned up temp directory: {self.temp_dir}")
+        except Exception as e:
+            logger.warning(f"Failed to clean up temp directory: {e}")
 
 session_manager = SessionManager()
 
@@ -307,7 +325,7 @@ def ca_chat(
     message: str,
     enable_redaction: bool,
     gentle_tone: bool,
-    log_path: str
+    use_temp_logging: bool  # New parameter to control logging location
 ) -> Tuple[List[Dict[str, str]], Dict[str, Any], Optional[str]]:
     """
     Enhanced chat handler with proper Gradio messages format and error handling.
@@ -365,17 +383,30 @@ def ca_chat(
         audit_event = make_audit_event(clean_message, found_redactions, compass_result, reply)
         audit_dict = asdict(audit_event)
         
-        if log_path:
+        download_path = None
+        
+        if use_temp_logging:
             try:
-                log_file = Path(log_path).expanduser()
-                log_file.parent.mkdir(parents=True, exist_ok=True)
-                with log_file.open("a", encoding="utf-8") as f:
+                # Use temporary directory that Gradio can access
+                temp_audit_file = session_manager.create_temp_audit_file()
+                with temp_audit_file.open("a", encoding="utf-8") as f:
                     f.write(json.dumps(audit_dict, ensure_ascii=False) + "\n")
-                logger.info(f"Audit event written to {log_file}")
+                download_path = str(temp_audit_file)
+                logger.info(f"Audit event written to temporary file: {temp_audit_file}")
             except Exception as e:
-                logger.error(f"Failed to write audit log: {e}")
+                logger.error(f"Failed to write temporary audit log: {e}")
+        else:
+            # Original logging to custom path (without download capability)
+            try:
+                log_path = Path("~/.outer_void/audit/blux_ca_demo.jsonl").expanduser()
+                log_path.parent.mkdir(parents=True, exist_ok=True)
+                with log_path.open("a", encoding="utf-8") as f:
+                    f.write(json.dumps(audit_dict, ensure_ascii=False) + "\n")
+                logger.info(f"Audit event written to persistent log: {log_path}")
+            except Exception as e:
+                logger.error(f"Failed to write persistent audit log: {e}")
 
-        return updated_history, audit_dict, str(log_file) if log_path and log_file.exists() else None
+        return updated_history, audit_dict, download_path
 
     except Exception as e:
         logger.error(f"Error in ca_chat: {e}")
@@ -440,6 +471,11 @@ def build_ui():
                         label="🌱 Gentle tone mode",
                         info="Softer, more supportive language"
                     )
+                    temp_logging = gr.Checkbox(
+                        True,
+                        label="📥 Enable file downloads",
+                        info="Allows downloading audit logs (uses temp files)"
+                    )
                 
                 with gr.Row():
                     send_btn = gr.Button("Send", variant="primary", size="lg")
@@ -465,41 +501,43 @@ def build_ui():
                     value={}
                 )
                 
-                log_file = gr.Textbox(
-                    value="~/.outer_void/audit/blux_ca_demo.jsonl",
-                    label="Audit Log Path",
-                    info="JSONL file for append-only logging"
-                )
-                
+                gr.Markdown("### 💾 Download Audit Log")
                 download = gr.File(
-                    label="Download Audit Log",
-                    visible=False
+                    label="Session Audit Log",
+                    visible=False,
+                    file_types=[".jsonl"],
+                    type="filepath"
                 )
 
         # Event handlers
-        def on_send(history, message, redact_enabled, gentle_enabled, log_path):
+        def on_send(history, message, redact_enabled, gentle_enabled, temp_logging_enabled):
             if not message or not message.strip():
                 gr.Warning("Please enter a message before sending.")
                 return history, {}, None, session_manager.get_session_info()
             
             new_history, audit_data, download_path = ca_chat(
-                history or [], message, redact_enabled, gentle_enabled, log_path
+                history or [], message, redact_enabled, gentle_enabled, temp_logging_enabled
             )
             
             session_info = session_manager.get_session_info()
-            download_component = download_path if download_path else None
             
-            return new_history, audit_data, download_component, session_info
+            # Show download button only if we have a downloadable file
+            download_visible = download_path is not None
+            download_value = download_path if download_path else None
+            
+            return new_history, audit_data, gr.update(visible=download_visible, value=download_value), session_info
 
         def on_clear():
-            session_manager.session_id = str(uuid.uuid4())
-            session_manager.start_time = time.time()
-            return [], {}, None, session_manager.get_session_info()
+            # Clean up old session and start new one
+            session_manager.cleanup()
+            session_manager.__init__()  # Reinitialize for new session
+            
+            return [], {}, gr.update(visible=False, value=None), session_manager.get_session_info()
 
         # Connect components
         send_btn.click(
             fn=on_send,
-            inputs=[chat, msg, redact_toggle, gentle_tone, log_file],
+            inputs=[chat, msg, redact_toggle, gentle_tone, temp_logging],
             outputs=[chat, audit_json, download, session_info]
         ).then(
             lambda: "",  # Clear message input
@@ -508,7 +546,7 @@ def build_ui():
 
         msg.submit(
             fn=on_send,
-            inputs=[chat, msg, redact_toggle, gentle_tone, log_file],
+            inputs=[chat, msg, redact_toggle, gentle_tone, temp_logging],
             outputs=[chat, audit_json, download, session_info]
         ).then(
             lambda: "",  # Clear message input
@@ -532,6 +570,11 @@ def build_ui():
         **Safety First**: Automatic crisis detection with immediate resource guidance
         
         **Privacy**: Optional redaction of personal information before processing
+        
+        ### 📁 File Downloads
+        
+        Enable "Download Audit Logs" to get a JSONL file containing all conversation events.
+        Files are stored in a temporary directory and cleaned up when you clear the history.
         """)
 
     return demo
@@ -548,13 +591,23 @@ if __name__ == "__main__":
         logger.error("Gradio not installed. Run: pip install gradio>=4.44.0")
         exit(1)
     
-    # Launch the application
-    demo = build_ui()
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False,
-        show_error=True,
-        debug=False,
-        ssr_mode=False
-    )
+    try:
+        # Launch the application with allowed paths for file downloads
+        demo = build_ui()
+        demo.launch(
+            server_name="0.0.0.0",
+            server_port=7860,
+            share=False,
+            show_error=True,
+            debug=False,
+            ssr_mode=False,
+            # Allow temp directory access for file downloads
+            allowed_paths=[session_manager.temp_dir]
+        )
+    except KeyboardInterrupt:
+        logger.info("Received interrupt signal - shutting down gracefully...")
+    except Exception as e:
+        logger.error(f"Failed to launch application: {e}")
+    finally:
+        # Clean up on exit
+        session_manager.cleanup()