Update app.py

app.py

@@ -80,12 +80,12 @@ def start_ssh_tunnel():
         else:
             logger.warning("Failed to start SSH tunnel. Falling back to Hyperbolic API.")
             use_fallback = True
-            tunnel_status = {"is_running": False, "message": "Failed to connect"}
+            tunnel_status = {"is_running": False, "message": "Connection failed"}
     
     except Exception as e:
         logger.error(f"Error starting SSH tunnel: {str(e)}")
         use_fallback = True
-        tunnel_status = {"is_running": False, "message": f"Error: {str(e)}"}
+        tunnel_status = {"is_running": False, "message": "Connection error"}
 
 def check_vllm_api_health():
     """
@@ -140,13 +140,15 @@ def monitor_tunnel():
                         use_fallback = True
                         tunnel_status = {
                             "is_running": False,
-                            "message": f"Tunnel connected but vLLM API unhealthy: {message}"
+                            "message": "Tunnel connected but vLLM API unhealthy"
                         }
                 else:
+                    # Log the actual error for troubleshooting but don't expose it in the UI
+                    logger.error(f"SSH tunnel disconnected: {ssh_status['error'] or 'Unknown error'}")
                     use_fallback = True
                     tunnel_status = {
                         "is_running": False,
-                        "message": f"Disconnected: {ssh_status['error'] or 'Unknown error'}"
+                        "message": "Disconnected - Check server status"
                     }
             else:
                 use_fallback = True
@@ -155,7 +157,7 @@ def monitor_tunnel():
         except Exception as e:
             logger.error(f"Error monitoring tunnel: {str(e)}")
             use_fallback = True
-            tunnel_status = {"is_running": False, "message": f"Monitoring error: {str(e)}"}
+            tunnel_status = {"is_running": False, "message": "Monitoring error"}
         
         time.sleep(5)  # Check every 5 seconds
 
@@ -382,12 +384,12 @@ def process_chat(message_dict, history):
             
             except Exception as fallback_error:
                 logger.error(f"Fallback API error: {str(fallback_error)}")
-                error_msg = f"Error with both primary and fallback APIs. Primary: {str(primary_error)}. Fallback: {str(fallback_error)}"
+                error_msg = "Error connecting to both primary and fallback APIs."
                 history.append({"role": "assistant", "content": error_msg})
                 return history
         else:
             # Already using fallback, just report the error
-            error_msg = f"An error occurred with the model: {str(primary_error)}"
+            error_msg = "An error occurred with the model service."
             history.append({"role": "assistant", "content": error_msg})
             return history