Update app.py

app.py

@@ -15,6 +15,7 @@ import logging
 import torch
 import gc
 import atexit
+import signal
 from diskcache import Cache
 from transformers import AutoTokenizer
 from datetime import datetime
@@ -278,7 +279,9 @@ class ClinicalOversightApp:
         self.agent = self._initialize_agent()
         self.text_processor = TextProcessor()
         self.file_processor = FileProcessor()
-        atexit.register(self.cleanup_resources)  # Register cleanup on exit
+        atexit.register(self.cleanup_resources)
+        signal.signal(signal.SIGTERM, self._signal_handler)
+        signal.signal(signal.SIGINT, self._signal_handler)
 
     def _initialize_agent(self):
         """Initialize the TxAgent with proper configuration"""
@@ -306,15 +309,26 @@ class ClinicalOversightApp:
         logger.info("AI Agent Ready")
         return agent
 
+    def _signal_handler(self, signum, frame):
+        """Handle termination signals"""
+        logger.info(f"Received signal {signum}, cleaning up...")
+        self.cleanup_resources()
+        sys.exit(0)
+
     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()
+        for _ in range(2):  # Retry to ensure cleanup
+            try:
+                if torch.distributed.is_initialized():
+                    logger.info("Destroying PyTorch distributed process group...")
+                    torch.distributed.destroy_process_group()
+                    break
+            except Exception as e:
+                logger.error(f"Cleanup error: {e}")
         log_system_resources("After Cleanup")
 
     def process_response_stream(self, prompt: str, history: List[dict]) -> Generator[dict, None, None]:
@@ -486,6 +500,7 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
             transition: all 0.3s ease;
             background: var(--message-bg);
             position: relative;
+            animation: messageFade 0.3s ease;
         }
         .message:hover {
             transform: translateY(-2px);
@@ -610,6 +625,9 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
             z-index: 1000;
             animation: fadeIn 0.3s ease;
         }
+        .markdown-tooltip, .file-tooltip {
+            display: block;
+        }
         .loading-spinner {
             position: absolute;
             bottom: 80px;
@@ -663,6 +681,10 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
             50% { width: 50%; }
             100% { width: 0; }
         }
+        @keyframes messageFade {
+            from { opacity: 0; transform: translateY(10px); }
+            to { opacity: 1; transform: translateY(0); }
+        }
         :root {
             --background: #ffffff;
             --text-color: #333333;
@@ -766,13 +788,16 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
                         "<div class='tooltip' data-tip='Upload patient records'>### 📎 Upload Records</div>",
                         elem_classes="markdown-tooltip"
                     )
+                    gr.HTML(
+                        "<div class='file-tooltip' data-tip='Select PDF, CSV, or Excel files'>"
+                    )
                     file_upload = gr.File(
                         file_types=[".pdf", ".csv", ".xls", ".xlsx"],
                         file_count="multiple",
                         label="Patient Records",
-                        elem_classes="tooltip",
-                        title="Select PDF, CSV, or Excel files"
+                        elem_classes="file-input"
                     )
+                    gr.HTML("</div>")
                     gr.Markdown(
                         "<div class='tooltip' data-tip='Summary of findings'>### 📝 Analysis Summary</div>",
                         elem_classes="markdown-tooltip"
@@ -785,13 +810,16 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
                         "<div class='tooltip' data-tip='Download full report'>### 📄 Full Report</div>",
                         elem_classes="markdown-tooltip"
                     )
+                    gr.HTML(
+                        "<div class='file-tooltip' data-tip='Download analysis report'>"
+                    )
                     download_output = gr.File(
                         label="Download Report",
                         visible=False,
                         interactive=False,
-                        elem_classes="tooltip",
-                        title="Download analysis report"
+                        elem_classes="file-output"
                     )
+                    gr.HTML("</div>")
                 
                 with gr.Row(elem_classes="input-container"):
                     msg_input = gr.Textbox(