Update app.py

app.py

@@ -14,6 +14,7 @@ import subprocess
 import logging
 import torch
 import gc
+import atexit
 from diskcache import Cache
 from transformers import AutoTokenizer
 from datetime import datetime
@@ -277,6 +278,7 @@ class ClinicalOversightApp:
         self.agent = self._initialize_agent()
         self.text_processor = TextProcessor()
         self.file_processor = FileProcessor()
+        atexit.register(self.cleanup_resources)  # Register cleanup on exit
 
     def _initialize_agent(self):
         """Initialize the TxAgent with proper configuration"""
@@ -307,11 +309,13 @@ class ClinicalOversightApp:
     def cleanup_resources(self):
         """Clean up GPU memory and collect garbage"""
         logger.info("Cleaning up resources...")
+        log_system_resources("Before Cleanup")
         torch.cuda.empty_cache()
         gc.collect()
         if torch.distributed.is_initialized():
             logger.info("Destroying PyTorch distributed process group...")
             torch.distributed.destroy_process_group()
+        log_system_resources("After Cleanup")
 
     def process_response_stream(self, prompt: str, history: List[dict]) -> Generator[dict, None, None]:
         """Stream the agent's response with proper formatting"""
@@ -592,7 +596,7 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
             position: relative;
         }
         .tooltip:hover::after {
-            content: attr(data-tooltip);
+            content: attr(data-tip);
             position: absolute;
             bottom: 100%;
             left: 50%;
@@ -758,27 +762,35 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
                     tools_button = gr.Button("📂 Tools", variant="secondary")
                 
                 with gr.Column(elem_classes="sidebar"):
-                    gr.Markdown("### 📎 Upload Records", elem_classes="tooltip", data_tooltip="Upload patient records")
+                    gr.Markdown(
+                        "<div class='tooltip' data-tip='Upload patient records'>### 📎 Upload Records</div>",
+                        elem_classes="markdown-tooltip"
+                    )
                     file_upload = gr.File(
                         file_types=[".pdf", ".csv", ".xls", ".xlsx"],
                         file_count="multiple",
                         label="Patient Records",
                         elem_classes="tooltip",
-                        data_tooltip="Select PDF, CSV, or Excel files"
+                        title="Select PDF, CSV, or Excel files"
+                    )
+                    gr.Markdown(
+                        "<div class='tooltip' data-tip='Summary of findings'>### 📝 Analysis Summary</div>",
+                        elem_classes="markdown-tooltip"
                     )
-                    gr.Markdown("### 📝 Analysis Summary", elem_classes="tooltip", data_tooltip="Summary of findings")
                     final_summary = gr.Markdown(
-                        "Analysis results will appear here...",
-                        elem_classes="tooltip",
-                        data_tooltip="View analysis results"
+                        "<div class='tooltip' data-tip='View analysis results'>Analysis results will appear here...</div>",
+                        elem_classes="markdown-tooltip"
+                    )
+                    gr.Markdown(
+                        "<div class='tooltip' data-tip='Download full report'>### 📄 Full Report</div>",
+                        elem_classes="markdown-tooltip"
                     )
-                    gr.Markdown("### 📄 Full Report", elem_classes="tooltip", data_tooltip="Download full report")
                     download_output = gr.File(
                         label="Download Report",
                         visible=False,
                         interactive=False,
                         elem_classes="tooltip",
-                        data_tooltip="Download analysis report"
+                        title="Download analysis report"
                     )
                 
                 with gr.Row(elem_classes="input-container"):
@@ -879,7 +891,8 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
                 
                 app.load(
                     fn=lambda: [
-                        [], None, "", "", None, {"visible": False}, "light", False, "🌙 Dark Mode"
+                        [], None, "<div class='tooltip' data-tip='View analysis results'>Analysis results will appear here...</div>", 
+                        "", None, {"visible": False}, "light", False, "🌙 Dark Mode"
                     ],
                     outputs=[chatbot, download_output, final_summary, msg_input, file_upload, progress_text, theme_state, sidebar_state, theme_button],
                     queue=False