Upd models loader #6 and UI model loader display

config.py

@@ -54,8 +54,8 @@ DESCRIPTION = """
 <p>📄 <strong>Document RAG:</strong> Answer based on uploaded medical documents</p>
 <p>🌐 <strong>Web Search:</strong> Fetch knowledge from reliable online medical resources</p>
 <p>🌍 <strong>Multi-language:</strong> Automatic translation for non-English queries</p>
-<p>Tips: Customise configurations, system prompt to see where the magic happens!</p>
-<p>Note: Case GPU aborted errors, please select another model or try again later!</p>
+<p>Tips: Customise configurations, system prompt to see the magic happens!</p>
+<p>Note: Case GPU aborted or MedSwin not ready, please select another model!</p>
 </center>
 """
 CSS = """
@@ -144,6 +144,8 @@ CSS = """
     font-weight: 500;
     background-color: #f5f5f5;
     border: 1px solid #e0e0e0;
+    overflow-y: auto;
+    max-height: 120px;
 }
 @media (min-width: 768px) {
     .main-container {

ui.py

@@ -194,6 +194,8 @@ def create_demo():
                         label="Model Status",
                         interactive=False,
                         visible=True,
+                        lines=3,
+                        max_lines=3,
                         elem_classes="model-status"
                     )
                     
@@ -342,14 +344,14 @@ def create_demo():
                                     logger.error(f"Error calling load_model_with_gpu: {e}")
                                     status_lines.append(f"⏳ MedSwin ({model_name}): loading...")
                         
-                        # TTS model status
+                        # TTS model status (only show if available or if there's an issue)
                         if TTS_AVAILABLE:
                             if config.global_tts_model is not None:
                                 status_lines.append("✅ TTS (maya1): loaded and ready")
                             else:
-                                status_lines.append("⚠️ TTS (maya1): not loaded")
-                        else:
-                            status_lines.append("❌ TTS: library not available")
+                                # TTS available but not loaded - optional feature
+                                pass  # Don't show if not loaded, it's optional
+                        # Don't show TTS status if library not available (it's optional)
                         
                         # ASR (Whisper) model status
                         if WHISPER_AVAILABLE:
@@ -382,14 +384,12 @@ def create_demo():
                         else:
                             status_lines.append(f"⚠️ MedSwin ({model_name}): not loaded")
                     
-                    # TTS model status
+                    # TTS model status (only show if available and loaded)
                     if TTS_AVAILABLE:
                         if config.global_tts_model is not None:
                             status_lines.append("✅ TTS (maya1): loaded and ready")
-                        else:
-                            status_lines.append("⚠️ TTS (maya1): not loaded")
-                    else:
-                        status_lines.append("❌ TTS: library not available")
+                        # Don't show if TTS library available but model not loaded (optional feature)
+                    # Don't show TTS status if library not available (it's optional)
                     
                     # ASR (Whisper) model status
                     if WHISPER_AVAILABLE:
@@ -441,7 +441,7 @@ def create_demo():
                             torch.cuda.empty_cache()
                             logger.debug("[STARTUP] Cleared GPU cache after medical model")
                         
-                        # Step 2: Load TTS model (maya1)
+                        # Step 2: Load TTS model (maya1) - optional, only show if loaded
                         if TTS_AVAILABLE:
                             logger.info("[STARTUP] Step 2/3: Loading TTS model (maya1)...")
                             try:
@@ -449,15 +449,10 @@ def create_demo():
                                 if config.global_tts_model is not None:
                                     status_messages.append("✅ TTS (maya1): loaded")
                                     logger.info("[STARTUP] ✅ TTS model loaded successfully!")
-                                else:
-                                    status_messages.append("⚠️ TTS (maya1): failed")
-                                    logger.warning("[STARTUP] ⚠️ TTS model failed to load")
+                                # Don't show status if TTS failed to load (it's optional)
                             except Exception as e:
-                                status_messages.append("❌ TTS (maya1): error")
-                                logger.error(f"[STARTUP] TTS model loading error: {e}")
-                        else:
-                            status_messages.append("❌ TTS: library not available")
-                            logger.warning("[STARTUP] TTS library not installed")
+                                logger.warning(f"[STARTUP] TTS model loading error (optional): {e}")
+                        # Don't show TTS status if library not available (it's optional)
                         
                         # Small delay to let GPU settle and clear cache
                         time.sleep(2)